zq/fix ascend index_select (DeepLink-org#832)

remove device_config of ascend index_select and add diopiIndexSelectBackward

Author: NeosZhang

impl/ascend/device_configs.py

@@ -657,30 +657,6 @@
         ),
     ),
 
-    'index_select': dict(
-        name=['index_select'],
-        tensor_para=dict(
-            args=[
-                {
-                    "ins": ['input'],
-                    "dtype": [Skip(np.float32),Skip(np.float64),Skip(np.float16),],
-                },
-            ]
-        ),
-    ),
-
-    'index_select_not_float': dict(
-        name=['index_select'],
-        tensor_para=dict(
-            args=[
-                {
-                    "ins": ['input'],
-                    "dtype": [Skip(np.int32),Skip(np.int16),Skip(np.int64),Skip(np.uint8),Skip(np.int8),Skip(np.bool_),],
-                },
-            ]
-        ),
-    ),
-
     'masked_scatter': dict(
         name=['masked_scatter'],
         tensor_para=dict(

impl/ascend/functions/index_select.cpp

@@ -15,5 +15,19 @@ diopiError_t diopiIndexSelect(diopiContextHandle_t ctx, diopiTensorHandle_t out,
     return diopiSuccess;
 }
 
+diopiError_t diopiIndexSelectBackward(diopiContextHandle_t ctx, diopiTensorHandle_t gradInput, diopiConstTensorHandle_t grad, diopiSize_t inputSizes,
+                                      int64_t dim, diopiConstTensorHandle_t index) {
+    AscendTensor gradInputAt(gradInput);
+    if (dim < 0) {
+        dim = dim + inputSizes.len;
+    }
+    std::vector<int64_t> dimVec({dim});
+    diopiSize_t dimInput = vectorToDiopiSize(dimVec);
+    diopiScalar_t scalarZero = constructDiopiScalarT(gradInputAt.dtype(), 0);
+    diopiFill(ctx, gradInput, &scalarZero);
+    AclOpRunner<3, 1>("InplaceIndexAdd", ctx).addInput(gradInput).addInput(index).addInput(grad).setAttr<int64_t>("axis", dim).addOutput(gradInput).run();
+    return diopiSuccess;
+}
+
 }  // namespace ascend
 }  // namespace impl