Associate KernelTree arguments with  AIR function arguments properly