Browse Source

complete the int64 support of gpu batchnorm fold2 grad operation

tags/v1.1.0
Payne 5 years ago
parent
commit
9cbb9bc428
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/ccsrc/backend/kernel_compiler/gpu/quant/batchnorm_fold2_grad_gpu_kernel.h

+ 1
- 1
mindspore/ccsrc/backend/kernel_compiler/gpu/quant/batchnorm_fold2_grad_gpu_kernel.h View File

@@ -114,7 +114,7 @@ class BatchNormFold2GradGpuKernel : public GpuKernel {
channel_ = input_shape[1]; channel_ = input_shape[1];
height_ = input_shape[2]; height_ = input_shape[2];
width_ = input_shape[3]; width_ = input_shape[3];
freeze_bn_ = GetValue<int32_t>(AnfAlgo::GetCNodePrimitive(kernel_node)->GetAttr("freeze_bn"));
freeze_bn_ = GetValue<int64_t>(AnfAlgo::GetCNodePrimitive(kernel_node)->GetAttr("freeze_bn"));


InitSizeLists(); InitSizeLists();
return true; return true;


Loading…
Cancel
Save