!9081 int32->int64 support

From: @yepei6 Reviewed-by: @kingxian,@zh_qh Signed-off-by: @kingxian
4 years ago · cf0bca06eb
parent e211e11b48 9cbb9bc428
commit cf0bca06eb
1 changed files with 1 additions and 1 deletions
--- a/mindspore/ccsrc/backend/kernel_compiler/gpu/quant/batchnorm_fold2_grad_gpu_kernel.h
+++ b/mindspore/ccsrc/backend/kernel_compiler/gpu/quant/batchnorm_fold2_grad_gpu_kernel.h
@ -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;