From 121615e2ed3ec70b40a5bdac183d4a7d5e43d410 Mon Sep 17 00:00:00 2001 From: Peilin Wang Date: Fri, 12 Mar 2021 16:14:22 -0500 Subject: [PATCH] remove int32 support for TensorScatterUpdate --- .../gpu/arrays/tensor_scatter_update_gpu_kernel.cc | 8 -------- 1 file changed, 8 deletions(-) diff --git a/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/tensor_scatter_update_gpu_kernel.cc b/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/tensor_scatter_update_gpu_kernel.cc index 70ac26aa97..f7ef765ab5 100644 --- a/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/tensor_scatter_update_gpu_kernel.cc +++ b/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/tensor_scatter_update_gpu_kernel.cc @@ -42,14 +42,6 @@ MS_REG_GPU_KERNEL_TWO(TensorScatterUpdate, .AddOutputAttr(kNumberTypeInt8), TensorScatterUpdateGpuFwdKernel, char, int) -MS_REG_GPU_KERNEL_TWO(TensorScatterUpdate, - KernelAttr() - .AddInputAttr(kNumberTypeInt32) - .AddInputAttr(kNumberTypeInt32) - .AddInputAttr(kNumberTypeInt32) - .AddOutputAttr(kNumberTypeInt32), - TensorScatterUpdateGpuFwdKernel, int, int) - MS_REG_GPU_KERNEL_TWO(TensorScatterUpdate, KernelAttr() .AddInputAttr(kNumberTypeUInt8)