mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

fix output_size zero bug.

This commit is contained in:
linqingke 2021-06-29 17:30:07 +08:00
parent cc8d760c8b
commit 1498953ea0
3 changed files with 0 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/gather_gpu_kernel.h
View File

@ -97,9 +97,6 @@ class GatherGpuFwdKernel : public GpuKernel {
return; return;
} }
size_t GetSize(const std::vector<size_t> &shape, const bool flag = true) const { size_t GetSize(const std::vector<size_t> &shape, const bool flag = true) const {
if (shape.size() == 0) {
return 0;
}
size_t result = flag ? sizeof(T) : sizeof(S); size_t result = flag ? sizeof(T) : sizeof(S);
for (size_t i = 0; i < shape.size(); i++) { for (size_t i = 0; i < shape.size(); i++) {
result *= shape[i]; result *= shape[i];

3
mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/gather_grad_gpu_kernel.h
View File

@ -98,9 +98,6 @@ class GatherGradGpuKernel : public GpuKernel {
return; return;
} }
size_t GetSize(const std::vector<size_t> &shape, const bool flag = true) const { size_t GetSize(const std::vector<size_t> &shape, const bool flag = true) const {
if (shape.size() == 0) {
return 0;
}
size_t result = flag ? sizeof(T) : sizeof(S); size_t result = flag ? sizeof(T) : sizeof(S);
for (size_t i = 0; i < shape.size(); i++) { for (size_t i = 0; i < shape.size(); i++) {
result *= shape[i]; result *= shape[i];

3
mindspore/ccsrc/backend/session/anf_runtime_algorithm.h
View File

@ -288,9 +288,6 @@ class AnfRuntimeAlgorithm {
// Calc tensor size in byte. // Calc tensor size in byte.
template <typename T> template <typename T>
static size_t TensorSizeInByte(const std::vector<size_t> &shape) { static size_t TensorSizeInByte(const std::vector<size_t> &shape) {
if (shape.size() == 0) {
return 0;
}
size_t result = sizeof(T); size_t result = sizeof(T);
for (size_t i = 0; i < shape.size(); i++) { for (size_t i = 0; i < shape.size(); i++) {
result *= shape[i]; result *= shape[i];