[INTEL_HPU] fix issue that graph recompile not triggered by index dtype change (#1861)

Author: JianyuLi01

backends/intel_hpu/kernels/index_select_kernel.cc

@@ -14,6 +14,7 @@
 
 #include "habanalabs/perf_lib_layer_params.h"
 #include "kernels/funcs.h"
+#include "kernels/hpu_funcs.h"
 #include "kernels/hpu_operator.h"
 #include "utils/utils.h"
 
@@ -85,12 +86,21 @@ void IndexSelectKernel(const Context& dev_ctx,
     dim += x.dims().size();
   }
 
+  std::string op_name = "IndexSelectKernel";
+  if (index.dtype() == phi::DataType::INT32) {
+    op_name += "_int32";
+  } else if (index.dtype() == phi::DataType::INT64) {
+    op_name += "_int64";
+  } else {
+    throw std::runtime_error(
+        "index_select supports only int64 and int32 for index!");
+  }
+
   OpCacheOperator op_info;
   IndexSelectParams params;
   params.params.axis = static_cast<int32_t>(x.dims().size()) - 1 - dim;
   std::vector<DIMS> inputs_dims = ct.GetDims();
-  op_info.prepareOpInfo<T, IndexSelectParams>(
-      "IndexSelectKernel", inputs_dims, &params);
+  op_info.prepareOpInfo<T, IndexSelectParams>(op_name, inputs_dims, &params);
   auto recipe = op_info.GetRecipe();
   if (recipe == nullptr) {
     IndexSelect op;

backends/intel_hpu/tests/unittests/test_index_select.py

@@ -114,5 +114,18 @@ def config(self):
         self.index_type = np.int64
 
 
+class TestMixOfInt32andInt64(unittest.TestCase):
+    def test_mix_int32_int64(self):
+        paddle.disable_static()
+        x = paddle.to_tensor([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
+        indices_32 = paddle.to_tensor([0, 2], dtype=paddle.int32)
+        result1 = paddle.index_select(x, axis=0, index=indices_32)
+
+        indices_64 = paddle.to_tensor([0, 2], dtype=paddle.int64)
+        result2 = paddle.index_select(x, axis=0, index=indices_64)
+        self.assertTrue(np.array_equal(result1.numpy(), result2.numpy()))
+        paddle.enable_static()
+
+
 if __name__ == "__main__":
     unittest.main()