From c77184b8ede019b04e961ef47ad999e4cc5cfae5 Mon Sep 17 00:00:00 2001 From: lowdy1 Date: Thu, 13 Aug 2026 08:01:36 +0000 Subject: [PATCH] tag conv1d as mix mode --- .../SeparateMemoryFromCompute/MarkGMLoadPass.cpp | 3 ++- third_party/ascend/lib/TritonToLinalg/TritonToLinalgPass.cpp | 4 ++++ 2 files changed, 6 insertions(+), 1 deletion(-) 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