unit test fix -2

f36f7f1e · rtmadduri · 8c569d96 · f36f7f1e
Commit f36f7f1e authored Jan 22, 2025 by rtmadduri
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 ...u/device/impl/device_grouped_gemm_xdl_splitk_cshuffle.hpp +6 -6

No files found.
--- a/include/ck/tensor_operation/gpu/device/impl/device_grouped_gemm_xdl_splitk_cshuffle.hpp
+++ b/include/ck/tensor_operation/gpu/device/impl/device_grouped_gemm_xdl_splitk_cshuffle.hpp
@@ -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 =