add emb cast in KTRegroupAsDict module (#3008) #1929

Job	Run time
build_test (linux.g5.12xlarge.nvidia.gpu, 3.9, py39, cu126) / linux-job	16m 42s
build_test (linux.g5.12xlarge.nvidia.gpu, 3.9, py39, cu128) / linux-job	17m 9s
build_test (linux.g5.12xlarge.nvidia.gpu, 3.10, py310, cu118) / linux-job	16m 19s
build_test (linux.g5.12xlarge.nvidia.gpu, 3.10, py310, cu126) / linux-job	16m 40s
build_test (linux.g5.12xlarge.nvidia.gpu, 3.10, py310, cu128) / linux-job	17m 7s
build_test (linux.g5.12xlarge.nvidia.gpu, 3.12, py312, cu126) / linux-job	16m 35s
build_test (linux.g5.12xlarge.nvidia.gpu, 3.11, py311, cu126) / linux-job	16m 38s
build_test (linux.g5.12xlarge.nvidia.gpu, 3.11, py311, cu118) / linux-job	16m 5s
build_test (linux.g5.12xlarge.nvidia.gpu, 3.9, py39, cu118) / linux-job	16m 7s
build_test (linux.g5.12xlarge.nvidia.gpu, 3.11, py311, cu128) / linux-job	16m 59s
build_test (linux.g5.12xlarge.nvidia.gpu, 3.12, py312, cu118) / linux-job	16m 4s
build_test (linux.g5.12xlarge.nvidia.gpu, 3.12, py312, cu128) / linux-job	16m 58s
	3h 19m 23s

Provide feedback