ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-04-20 06:49:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor: remove gemm preshuffle pipeline v1 by removing all references from codebase (#3132)

Browse Source 
* test: temporarily disable flaky test_ck_tile_moe_sorting_2d_buf

* refactor: deprecate gemm preshuffle pipeline v1 by removing all references from codebase

* Revert "test: temporarily disable flaky test_ck_tile_moe_sorting_2d_buf"

This reverts commit 573c08a085.
This commit is contained in:
Aviral Goel
2025-11-02 00:06:28 -04:00
committed by GitHub
parent 45be741586
commit 73f637894d
7 changed files with 5 additions and 558 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tile_engine/ops/gemm_preshuffle/gemm_preshuffle_common.hpp
View File

@@ -78,7 +78,7 @@ constexpr auto is_row_major(Layout)
// Structure to hold kernel traits for dispatcher
struct KernelTraits
{
std::string pipeline; // preshufflev1, preshufflev2
std::string pipeline; // preshufflev2
std::string scheduler; // intrawave, interwave, default
std::string epilogue; // cshuffle, default
bool pad_m;
@@ -105,11 +105,7 @@ inline KernelTraits extract_traits_from_name(const std::string& kernel_name)
KernelTraits traits;
// Extract pipeline
if(kernel_name.find("preshufflev1") != std::string::npos)
{
traits.pipeline = "preshufflev1";
}
else if(kernel_name.find("preshufflev2") != std::string::npos)
if(kernel_name.find("preshufflev2") != std::string::npos)
{
traits.pipeline = "preshufflev2";
}

2
tile_engine/ops/gemm_preshuffle/gemm_preshuffle_instance_builder.py
View File

@@ -357,13 +357,11 @@ class GemmPreshuffleKernelBuilder:
# Map pipeline names to the correct pipeline implementation
pipeline_impl_map = {
"preshufflev1": "ck_tile::WeightPreshufflePipelineAGmemBGmemCRegV1",
"preshufflev2": "ck_tile::WeightPreshufflePipelineAGmemBGmemCRegV2",
}
# Map pipeline names to base pipeline for hot loop detection
base_pipeline_map = {
"preshufflev1": "ck_tile::BaseWeightPreshufflePipelineAGmemBGmemCRegV1",
"preshufflev2": "ck_tile::BaseWeightPreshufflePipelineAGmemBGmemCRegV2",
}