diff --git a/third_party/ascend/lib/DynamicCVPipeline/SeparateMemoryFromCompute/MarkGMLoadPass.cpp b/third_party/ascend/lib/DynamicCVPipeline/SeparateMemoryFromCompute/MarkGMLoadPass.cpp index 6fb066fa71..3e40eab748 100644 --- a/third_party/ascend/lib/DynamicCVPipeline/SeparateMemoryFromCompute/MarkGMLoadPass.cpp +++ b/third_party/ascend/lib/DynamicCVPipeline/SeparateMemoryFromCompute/MarkGMLoadPass.cpp @@ -306,7 +306,8 @@ void MarkGMLoadPass::runOnOperation() { // Skip marking for the sdpa infer kernel. bool isSdpaInferKernel = false; module.walk([&](func::FuncOp funcOp) -> WalkResult { - if (funcOp.getSymName() == "_sdpa_infer_kernel" || funcOp.getSymName() == "kernel_sdpa_fwd") { + if (funcOp.getSymName() == "_sdpa_infer_kernel" || + funcOp.getSymName() == "kernel_sdpa_fwd") { isSdpaInferKernel = true; return WalkResult::interrupt(); } diff --git a/third_party/ascend/lib/TritonToLinalg/TritonToLinalgPass.cpp b/third_party/ascend/lib/TritonToLinalg/TritonToLinalgPass.cpp index 6483b91bf8..bfa7176184 100644 --- a/third_party/ascend/lib/TritonToLinalg/TritonToLinalgPass.cpp +++ b/third_party/ascend/lib/TritonToLinalg/TritonToLinalgPass.cpp @@ -949,6 +949,10 @@ void TritonToLinalgPass::runOnOperation() { existDot = true; return WalkResult::interrupt(); }); + moduleOp.walk([&](hfusion::Conv1DOp conv1dOp) { + existDot = true; + return WalkResult::interrupt(); + }); existDotFlag = existDot; // NOTE: existSIMTOp is intentionally computed AFTER