Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!33553 Fix bug for cuda10 

Merge pull request !33553 from hezhenhao1/fix_bug
This commit is contained in:
i-robot 2022-04-26 03:28:38 +00:00 committed by Gitee
parent 76f372ac4b d7e8fd54b5
commit 29fad8ef0a
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/cum_minmax_impl.cu
View File

@ -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_;