Skip to content

Commit c2e7103

Browse files
olivervncinsanecoder-zzj
authored andcommitted
[ssbuffer](fix) Remove fallback by name for certain kernels
Signed-off-by: chenwuyang <chenwuyang1@huawei.com>
1 parent 5930ffd commit c2e7103

2 files changed

Lines changed: 7 additions & 12 deletions

File tree

third_party/ascend/lib/DynamicCVPipeline/AnalyzeDataFlow/AnalyzeName.cpp

Lines changed: 6 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -41,15 +41,12 @@ using namespace triton;
4141
namespace {
4242

4343
static constexpr llvm::StringLiteral interceptrFunc[]{
44-
"chunk_abc_bwd_kernel_dh",
45-
"flash_varlen_fwd_kernel",
46-
"chunk_gsa_bwd_k_kernel_dqkvg",
47-
"_jagged_flash_attention_bwd_basic_kernel",
48-
"triton_flash_mla_sparse_fwd",
49-
"_sparse_decode_kernel",
50-
"chunk_gsa_fwd_k_kernel_intra",
51-
"_sparse_decode_model1_kernel",
52-
"sparse_flash_attention_grad_kernel"};
44+
"chunk_abc_bwd_kernel_dh", "flash_varlen_fwd_kernel",
45+
"chunk_gsa_bwd_k_kernel_dqkvg", "_jagged_flash_attention_bwd_basic_kernel",
46+
47+
"_sparse_decode_kernel", "chunk_gsa_fwd_k_kernel_intra",
48+
49+
"_sparse_decode_model1_kernel", "sparse_flash_attention_grad_kernel"};
5350

5451
static LogicalResult verifyFuncNames(ModuleOp module) {
5552
bool intercepted = false;

third_party/ascend/lib/DynamicCVPipeline/Common/Utils.cpp

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -213,9 +213,7 @@ CoreType getCoreTypeOfSimpleOpOrCf(Operation *op) {
213213
"backward_dkdv",
214214
"pcb10_tc01_kernel",
215215
"chunk_ttt_linear_fwd_kernel_h",
216-
"chunk_ttt_linear_bwd_kernel_h",
217-
"chunk_gated_delta_rule_bwd_kernel_dhu_k128_blockdim128",
218-
"chunk_gated_delta_rule_fwd_kernel_h_k128_blockdim128"};
216+
"chunk_ttt_linear_bwd_kernel_h"};
219217
if (llvm::is_contained(regionalDisabledOps, funcOp.getSymName())) {
220218
return CoreType::UNDETERMINED;
221219
}

0 commit comments

Comments
 (0)