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

!47447 Bugfix for the lstm gpu kernels 

Merge pull request !47447 from chengang/bugfix_lstm
This commit is contained in:
i-robot 2023-01-04 01:23:59 +00:00 committed by Gitee
parent ffad43dbca b59c188b8c
commit 517021f548
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
6 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
mindspore/ccsrc/plugin/device/gpu/kernel/nn/lstm_gpu_kernel.cc
View File

@ -65,6 +65,7 @@ int LstmGpuKernelMod::Resize(const BaseOperatorPtr &base_operator, const std::ve
if (auto ret = KernelMod::Resize(base_operator, inputs, outputs, inputsOnHost); ret != KRET_OK) {
return ret;
}
DestroyTensorDescGrp();
auto input_shape = inputs[kIndex0]->GetShapeVector();
is_null_input_ = CHECK_SHAPE_NULL(input_shape, kernel_name_, "input") ||

1
mindspore/ccsrc/plugin/device/gpu/kernel/nn/lstm_gpu_kernel.h
View File

@ -84,7 +84,6 @@ class LstmGpuKernelMod : public NativeGpuKernelMod {
}
void CreateTensorDescGrp() {
DestroyTensorDescGrp();
int x_dims[3]{batch_size_, input_size_, 1};
int y_dims[3]{batch_size_, hidden_size_ * (bidirectional_ ? 2 : 1), 1};

1
mindspore/ccsrc/plugin/device/gpu/kernel/nn/lstm_grad_data_gpu_kernel.cc
View File

@ -56,6 +56,7 @@ int LstmGradDataGpuKernelMod::Resize(const BaseOperatorPtr &base_operator, const
if (auto ret = KernelMod::Resize(base_operator, inputs, outputs, inputsOnHost); ret != KRET_OK) {
return ret;
}
DestroyTensorDescGrp();
auto input_shape = inputs[kIndex0]->GetShapeVector();
is_null_input_ = CHECK_SHAPE_NULL(input_shape, kernel_name_, "input");

1
mindspore/ccsrc/plugin/device/gpu/kernel/nn/lstm_grad_data_gpu_kernel.h
View File

@ -134,7 +134,6 @@ class LstmGradDataGpuKernelMod : public NativeGpuKernelMod {
}
void CreateTensorDescGrp() {
DestroyTensorDescGrp();
int x_dims[3]{batch_size_, input_size_, 1};
int y_dims[3]{batch_size_, hidden_size_ * (bidirectional_ ? 2 : 1), 1};

1
mindspore/ccsrc/plugin/device/gpu/kernel/nn/lstm_grad_weight_gpu_kernel.cc
View File

@ -56,6 +56,7 @@ int LstmGradWeightGpuKernelMod::Resize(const BaseOperatorPtr &base_operator, con
if (auto ret = KernelMod::Resize(base_operator, inputs, outputs, inputsOnHost); ret != KRET_OK) {
return ret;
}
DestroyTensorDescGrp();
auto input_shape = inputs[kIndex0]->GetShapeVector();
is_null_input_ = CHECK_SHAPE_NULL(input_shape, kernel_name_, "input");

1
mindspore/ccsrc/plugin/device/gpu/kernel/nn/lstm_grad_weight_gpu_kernel.h
View File

@ -111,7 +111,6 @@ class LstmGradWeightGpuKernelMod : public NativeGpuKernelMod {
}
void CreateTensorDescGrp() {
DestroyTensorDescGrp();
int x_dims[3]{batch_size_, input_size_, 1};
int y_dims[3]{batch_size_, hidden_size_ * (bidirectional_ ? 2 : 1), 1};