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

fixed softmax bug

This commit is contained in:
BokaiLi 2022-11-01 00:00:22 +08:00
parent 8ad405032d
commit a03c54a2ea
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/sparse/sparse_matrix_softmax_gpu_kernel.cc
View File

@ -68,7 +68,7 @@ int SparseMatrixSoftmaxGpuKernelMod::Resize(const BaseOperatorPtr &base_operator
input_row_pointers_ = std::vector<size_t>(inputs.at(kIndex2)->GetDeviceShapeAdaptively().begin(),
inputs.at(kIndex2)->GetDeviceShapeAdaptively().end());
row_pointers_elements_ =
std::accumulate(input_row_pointers_.begin(), input_row_pointers_.end(), 1, std::multiplies<size_t>());
std::accumulate(input_row_pointers_.begin(), input_row_pointers_.end(), 1, std::multiplies<size_t>()) - 1;
input_col_indices_ = std::vector<size_t>(inputs.at(kIndex3)->GetDeviceShapeAdaptively().begin(),
inputs.at(kIndex3)->GetDeviceShapeAdaptively().end());