Browse Source

!9081 int32->int64 support

From: @yepei6
Reviewed-by: @kingxian,@zh_qh
Signed-off-by: @kingxian
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
cf0bca06eb
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];
height_ = input_shape[2];
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();
return true;


Loading…
Cancel
Save