Add support for new KK ConsumerIR