Tweaked IR to support multiple tasks