update async memset

47d649a7 · carlushuang · e8490111 · 47d649a7
Commit 47d649a7 authored May 30, 2023 by carlushuang
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 3 deletions

include/ck/tensor_operation/gpu/device/impl/device_gemm_xdl_streamk.hpp ...sor_operation/gpu/device/impl/device_gemm_xdl_streamk.hpp +4 -3

No files found.
--- a/include/ck/tensor_operation/gpu/device/impl/device_gemm_xdl_streamk.hpp
+++ b/include/ck/tensor_operation/gpu/device/impl/device_gemm_xdl_streamk.hpp
@@ -173,9 +173,10 @@ struct DeviceGemmXdlStreamK : public DeviceGemmStreamK<ALayout,
                                                sizeof(typename GridwiseGemm::FloatAcc));
                auto preprocess = [&]() {
                    hipGetErrorString(
-                        hipMemset(workspace_semaphore,
-                                  0,
-                                  karg.block_mapping.get_workspace_size_for_semaphore()));
+                        hipMemsetAsync(workspace_semaphore,
+                                       0,
+                                       karg.block_mapping.get_workspace_size_for_semaphore(),
+                                       stream_config.stream_id_));
                };

                ave_time = launch_and_time_kernel_with_preprocess(stream_config,