Browse Source

CUDNN STATUS NOT SUPPORTED

pull/14002/head
dingpeifei 4 years ago
parent
commit
3ca1b4fc81
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      mindspore/ccsrc/backend/kernel_compiler/gpu/nn/batch_norm_gpu_kernel.h

+ 2
- 2
mindspore/ccsrc/backend/kernel_compiler/gpu/nn/batch_norm_gpu_kernel.h View File

@@ -96,6 +96,7 @@ class BatchNormGpuKernel : public GpuKernel {
} }


InitResource(); InitResource();
is_train_ = GetAttr<bool>(kernel_node, "is_training");
if (is_train_) { if (is_train_) {
mode_ = CUDNN_BATCHNORM_SPATIAL_PERSISTENT; mode_ = CUDNN_BATCHNORM_SPATIAL_PERSISTENT;
} else { } else {
@@ -133,7 +134,6 @@ class BatchNormGpuKernel : public GpuKernel {
} }
SetTensorDescriptor(format, shape); SetTensorDescriptor(format, shape);
InitSizeLists(); InitSizeLists();
is_train_ = GetAttr<bool>(kernel_node, "is_training");
return true; return true;
} }


@@ -229,8 +229,8 @@ class BatchNormGpuKernel : public GpuKernel {
} }


output_size_list_.push_back(output_size_); // output output_size_list_.push_back(output_size_); // output
output_size_list_.push_back(reserve_size_); // reserve space
output_size_list_.push_back(para_size_); // save scale output_size_list_.push_back(para_size_); // save scale
output_size_list_.push_back(reserve_size_); // reserve space
output_size_list_.push_back(para_size_); // save mean output_size_list_.push_back(para_size_); // save mean
output_size_list_.push_back(para_size_); // save variance output_size_list_.push_back(para_size_); // save variance




Loading…
Cancel
Save