Skip to content

Commit 73f6378

Browse files
refactor: remove gemm preshuffle pipeline v1 by removing all references from codebase (#3132)
* 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 573c08a.
1 parent 45be741 commit 73f6378

File tree

7 files changed

+5
-558
lines changed

7 files changed

+5
-558
lines changed

example/ck_tile/03_gemm/gemm_utils.hpp

Lines changed: 1 addition & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -17,8 +17,7 @@
1717
#define CK_TILE_PIPELINE_COMPUTE_V4 3
1818
#define CK_TILE_PIPELINE_COMPUTE_V5 4
1919
#define CK_TILE_PIPELINE_COMPUTE_V6 5
20-
#define CK_TILE_PIPELINE_PRESHUFFLE_V1 6
21-
#define CK_TILE_PIPELINE_PRESHUFFLE_V2 7
20+
#define CK_TILE_PIPELINE_PRESHUFFLE_V2 6
2221

2322
template <typename PrecType, ck_tile::index_t M_Warp_Tile>
2423
constexpr ck_tile::index_t get_k_warp_tile()
@@ -514,16 +513,6 @@ struct PipelineTypeTraits<CK_TILE_PIPELINE_COMPUTE_V6>
514513
using UniversalGemmPipeline = ck_tile::BaseGemmPipelineAgBgCrCompV6<PipelineProblem>;
515514
};
516515

517-
template <>
518-
struct PipelineTypeTraits<CK_TILE_PIPELINE_PRESHUFFLE_V1>
519-
{
520-
template <typename PipelineProblem>
521-
using GemmPipeline = ck_tile::WeightPreshufflePipelineAGmemBGmemCRegV1<PipelineProblem>;
522-
template <typename PipelineProblem>
523-
using UniversalGemmPipeline =
524-
ck_tile::BaseWeightPreshufflePipelineAGmemBGmemCRegV1<PipelineProblem>;
525-
};
526-
527516
template <>
528517
struct PipelineTypeTraits<CK_TILE_PIPELINE_PRESHUFFLE_V2>
529518
{

include/ck_tile/ops/gemm.hpp

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -59,7 +59,6 @@
5959
#include "ck_tile/ops/gemm/pipeline/tile_gemm_shape.hpp"
6060
#include "ck_tile/ops/gemm/pipeline/tile_gemm_traits.hpp"
6161
#include "ck_tile/ops/gemm/pipeline/wp_pipeline_agmem_bgmem_creg_base_policy.hpp"
62-
#include "ck_tile/ops/gemm/pipeline/wp_pipeline_agmem_bgmem_creg_v1.hpp"
6362
#include "ck_tile/ops/gemm/pipeline/wp_pipeline_agmem_bgmem_creg_v2.hpp"
6463
#include "ck_tile/ops/gemm/warp/warp_gemm.hpp"
6564
#include "ck_tile/ops/gemm/warp/warp_gemm_attribute_mfma.hpp"

0 commit comments

Comments
 (0)