From 5e0bc410b399dd17ee39afe9daa34ab234f2e9c0 Mon Sep 17 00:00:00 2001 From: looop5 Date: Mon, 16 Aug 2021 15:44:32 +0800 Subject: [PATCH] fix workspace address type --- .../ccsrc/backend/kernel_compiler/akg/gpu/akg_gpu_kernel_mod.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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),