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

!48167 fix a bug that transpose gpu op have memory leak 

Merge pull request !48167 from hanhuifeng/transpose_gpu_memory_leak
This commit is contained in:
i-robot 2023-01-30 02:59:52 +00:00 committed by Gitee
parent c67e759ffc ea29a90eb2
commit 226c57d959
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
mindspore/ccsrc/plugin/device/gpu/kernel/arrays/transpose_gpu_kernel.cc
View File

@ -116,6 +116,7 @@ bool TransposeGpuKernelMod::LaunchKernel(const std::vector<AddressPtr> &inputs,
}
void TransposeGpuKernelMod::GetPermValue(const std::vector<int64_t> &perm) {
input_perm_.clear();
for (size_t j = 0; j < perm.size(); j++) {
auto p = (perm[j] >= 0) ? perm[j] : (perm.size() + perm[j]);
if (p < 0) {