Skip to content

GitLab

"sims/git@developer.sourcefind.cn:cnjsdfcy/simbricks.git" did not exist on "2ca4e962c9827b7a633766c2acbf9b422de7a576"
Commit 437d35a2 authored by Qianfeng Zhang's avatar Qianfeng Zhang
Browse files

Remove the printing in device operator

parent 8301d8dd
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 3 deletions
include/ck/tensor_operation/gpu/device/impl/device_grouped_mha_fwd_xdl_cshuffle_v2.hpp
View file @ 437d35a2
...@@ -919,8 +919,6 @@ struct DeviceGroupedMultiheadAttentionForward_Xdl_CShuffle_V2 ...@@ -919,8 +919,6 @@ struct DeviceGroupedMultiheadAttentionForward_Xdl_CShuffle_V2
if(status == hipStreamCaptureStatusActive) if(status == hipStreamCaptureStatusActive)
{ {
std::cout << " Inside hipStreamCapturing ..." << std::endl;
size_t copy_size = arg.group_kernel_args_.size() * sizeof(GroupKernelArg); size_t copy_size = arg.group_kernel_args_.size() * sizeof(GroupKernelArg);
// ToDO: when to release this memory buffer? // ToDO: when to release this memory buffer?
...@@ -936,7 +934,6 @@ struct DeviceGroupedMultiheadAttentionForward_Xdl_CShuffle_V2 ...@@ -936,7 +934,6 @@ struct DeviceGroupedMultiheadAttentionForward_Xdl_CShuffle_V2
} }
else else
{ {
std::cout << " Outside hipStreamCapturing ..." << std::endl;
HIP_CHECK_ERROR( HIP_CHECK_ERROR(
hipMemcpyAsync(arg.p_workspace_, hipMemcpyAsync(arg.p_workspace_,
arg.group_kernel_args_.data(), arg.group_kernel_args_.data(),
......
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