Skip to content

GitLab

"git@developer.sourcefind.cn:wangsen/mineru.git" did not exist on "03ea29bde07cb91bded4d0acbd0c734de10bc448"
Commit 47d649a7 authored by carlushuang's avatar carlushuang
Browse files

update async memset

parent e8490111
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
View file @ 47d649a7
...@@ -173,9 +173,10 @@ struct DeviceGemmXdlStreamK : public DeviceGemmStreamK<ALayout, ...@@ -173,9 +173,10 @@ struct DeviceGemmXdlStreamK : public DeviceGemmStreamK<ALayout,
sizeof(typename GridwiseGemm::FloatAcc)); sizeof(typename GridwiseGemm::FloatAcc));
auto preprocess = [&]() { auto preprocess = [&]() {
hipGetErrorString( hipGetErrorString(
hipMemset(workspace_semaphore, hipMemsetAsync(workspace_semaphore,
0, 0,
karg.block_mapping.get_workspace_size_for_semaphore())); karg.block_mapping.get_workspace_size_for_semaphore(),
stream_config.stream_id_));
}; };
ave_time = launch_and_time_kernel_with_preprocess(stream_config, ave_time = launch_and_time_kernel_with_preprocess(stream_config,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment