!8282 Add GPU SparseGatherV2 kernel and support dynamic shape

From: @TFbunny Reviewed-by: @robingrosman,@tom__chen Signed-off-by: @tom__chen
4 years ago · d6da9129da
parent 5caded733e 7f769568a6
commit d6da9129da
2 changed files with 939 additions and 1 deletions
--- a/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/gatherv2_gpu_kernel.cc
+++ b/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/gatherv2_gpu_kernel.cc
@ -26,7 +26,6 @@ MS_REG_GPU_KERNEL_TWO(
  GatherV2,
  KernelAttr().AddInputAttr(kNumberTypeFloat16).AddInputAttr(kNumberTypeInt32).AddOutputAttr(kNumberTypeFloat16),
  GatherV2GpuFwdKernel, half, int)
-
 MS_REG_GPU_KERNEL_TWO(
  SparseGatherV2,
  KernelAttr().AddInputAttr(kNumberTypeFloat32).AddInputAttr(kNumberTypeInt32).AddOutputAttr(kNumberTypeFloat32),
--- a/tests/st/ops/gpu/test_sparse_gather_v2_op.py
+++ b/tests/st/ops/gpu/test_sparse_gather_v2_op.py