Skip to content

GitLab

"docs/source/en/vscode:/vscode.git/clone" did not exist on "3f7c3511dcc95e5bb9fd53399dfc4eb655e1d6fd"
Commit f36f7f1e authored by rtmadduri's avatar rtmadduri
Browse files

unit test fix -2

parent 8c569d96
Hide whitespace changes
Inline Side-by-side
Showing with 6 additions and 6 deletions
include/ck/tensor_operation/gpu/device/impl/device_grouped_gemm_xdl_splitk_cshuffle.hpp
View file @ f36f7f1e
...@@ -430,12 +430,12 @@ struct DeviceGroupedGemmXdlSplitKCShuffle : public DeviceGroupedGemmSplitK<ALayo ...@@ -430,12 +430,12 @@ struct DeviceGroupedGemmXdlSplitKCShuffle : public DeviceGroupedGemmSplitK<ALayo
} }
} }
ave_time = launch_and_time_kernel(stream_config, for(const auto& trans_arg : arg.gemm_kernel_args_)
kernel, {
dim3(arg.grid_size_), const auto& karg = trans_arg.karg_;
dim3(BlockSize), ave_time += launch_and_time_kernel(
0, stream_config, kernel, dim3(arg.grid_size_), dim3(BlockSize), 0, karg_);
arg.gemm_kernel_args_[0].karg_); }
}; };
constexpr index_t minimum_occupancy = constexpr index_t minimum_occupancy =
......
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