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

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

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(reserve_size_); // reserve space
output_size_list_.push_back(para_size_); // save mean
output_size_list_.push_back(para_size_); // save variance



Loading…
Cancel
Save