From 9cbb9bc428fda69dce50a13631464ba2a85ee5c8 Mon Sep 17 00:00:00 2001
From: Payne <payneye93@gmail.com>
Date: Fri, 27 Nov 2020 17:34:13 +0800
Subject: [PATCH] complete the int64 support of gpu batchnorm fold2 grad
 operation

---
 .../kernel_compiler/gpu/quant/batchnorm_fold2_grad_gpu_kernel.h | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git 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
index 61abfaf113..6f060ddf4c 100644
--- 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;