File tree Expand file tree Collapse file tree
third_party/ascend/lib/DynamicCVPipeline Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -41,15 +41,12 @@ using namespace triton;
4141namespace {
4242
4343static 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
5451static LogicalResult verifyFuncNames (ModuleOp module ) {
5552 bool intercepted = false ;
Original file line number Diff line number Diff 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 }
You can’t perform that action at this time.
0 commit comments