diff --git a/mace/core/runtime/opencl/opencl_runtime.cc b/mace/core/runtime/opencl/opencl_runtime.cc
index 37b4c59ccdb21d8c57e68f50a18001f7bc710fa4..b39e28b8178e16d1b3959c441fb3d31e0421c3f0 100644
--- a/mace/core/runtime/opencl/opencl_runtime.cc
+++ b/mace/core/runtime/opencl/opencl_runtime.cc
@@ -337,9 +337,9 @@ uint32_t OpenCLRuntime::GetKernelMaxWorkGroupSize(const cl::Kernel &kernel) {
 
 // TODO(liuqi): not compatible with mali gpu.
 uint32_t OpenCLRuntime::GetKernelWaveSize(const cl::Kernel &kernel) {
-  unsigned long long size = 0;
+  uint32_t size = 0;
   kernel.getWorkGroupInfo(*device_, CL_KERNEL_WAVE_SIZE_QCOM, &size);
-  return static_cast<uint32_t>(size);
+  return size;
 }
 
 }  // namespace mace