diff --git a/mindspore/ccsrc/backend/kernel_compiler/akg/gpu/akg_gpu_kernel_mod.cc b/mindspore/ccsrc/backend/kernel_compiler/akg/gpu/akg_gpu_kernel_mod.cc index fdfc9791caf..4ed0d553340 100644 --- a/mindspore/ccsrc/backend/kernel_compiler/akg/gpu/akg_gpu_kernel_mod.cc +++ b/mindspore/ccsrc/backend/kernel_compiler/akg/gpu/akg_gpu_kernel_mod.cc @@ -125,7 +125,7 @@ bool GpuKernelMod::Launch(const std::vector &inputs, const std::vect [](const AddressPtr &output) -> void * { return reinterpret_cast(&(output->addr)); }); if (!workspace.empty()) { (void)std::transform(std::begin(workspace), std::end(workspace), std::back_inserter(runtimeargs), - [](const AddressPtr &addr) -> void * { return addr->addr; }); + [](const AddressPtr &addr) -> void * { return reinterpret_cast(&(addr->addr)); }); } result = cuLaunchKernel(kernel_addr, thread_info[0], thread_info[1], thread_info[2], thread_info[3], thread_info[4], thread_info[5], 0, reinterpret_cast(stream_ptr),