From d7e8fd54b50e63c5d71d4f4779ad0ed7bcc93c2f Mon Sep 17 00:00:00 2001 From: hezhenhao1 Date: Tue, 26 Apr 2022 09:40:55 +0800 Subject: [PATCH] Fix bug for cuda10 --- .../device/gpu/kernel/cuda_impl/cuda_ops/cum_minmax_impl.cu | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/cum_minmax_impl.cu b/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/cum_minmax_impl.cu index a2e8ce165c6..a32e91fafdd 100644 --- a/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/cum_minmax_impl.cu +++ b/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/cum_minmax_impl.cu @@ -38,7 +38,7 @@ struct binary_op { size_t inner_size_; OP op; - __thrust_exec_check_disable__ __device__ size_t operator()(const size_t &lhs, const size_t &rhs) const { + __device__ size_t operator()(const size_t &lhs, const size_t &rhs) const { if (rhs % axis_size_) { size_t batch_idx = rhs / axis_size_; size_t axis_idx = rhs - batch_idx * axis_size_;