[mlir] fix the types used during the generation of the kernel param array