optimization of index select

Zjq9409 · Zjq9409 · commit 04afbfd00e15 · 2021-07-07T12:41:26.000Z
diff --git a/paddle/fluid/operators/index_select_op.h b/paddle/fluid/operators/index_select_op.h
@@ -196,8 +196,10 @@ void IndexSelectGradInner(const framework::ExecutionContext& context,
       auto src = input_data + input_start_offset + j * slice_size;
       auto p_out = p_output + output_start_offset + index_value * slice_size;
       auto dst = out_data + output_start_offset + index_value * slice_size;
+#if ((!defined __NVCC__) && (!defined __HIPCC__))
       IndexSelectAdd<DeviceContext, T> index_select_add;
       index_select_add(context, slice_size, src, p_out, dst);
+#endif
     }
   }
   x_grad->Resize(output_dim);

Original file line number	Diff line number	Diff line change
`@@ -196,8 +196,10 @@ void IndexSelectGradInner(const framework::ExecutionContext& context,`
`196`	`196`	`auto src = input_data + input_start_offset + j * slice_size;`
`197`	`197`	`auto p_out = p_output + output_start_offset + index_value * slice_size;`
`198`	`198`	`auto dst = out_data + output_start_offset + index_value * slice_size;`
	`199`	`+#if ((!defined __NVCC__) && (!defined __HIPCC__))`
`199`	`200`	`IndexSelectAdd<DeviceContext, T> index_select_add;`
`200`	`201`	`index_select_add(context, slice_size, src, p_out, dst);`
	`202`	`+#endif`
`201`	`203`	`}`
`202`	`204`	`}`
`203`	`205`	`x_grad->Resize(output_dim);`