From 8aa1b4705a09614dd91b375f05002563eb1a1b52 Mon Sep 17 00:00:00 2001
From: james1314520 <2435883077@qq,com>
Date: Tue, 21 Apr 2026 15:55:42 +0800
Subject: [PATCH] fix the rsqrtOp in UnaryVecOp

---
 src/target/codegen_npuir_dev.cc                             | 6 +++---
 testing/npuir/{linalg_ops => hfusion_ops}/test_rsqrt_dev.py | 0
 2 files changed, 3 insertions(+), 3 deletions(-)
 rename testing/npuir/{linalg_ops => hfusion_ops}/test_rsqrt_dev.py (100%)

diff --git a/src/target/codegen_npuir_dev.cc b/src/target/codegen_npuir_dev.cc
index e9f0329e5..b01117dbf 100644
--- a/src/target/codegen_npuir_dev.cc
+++ b/src/target/codegen_npuir_dev.cc
@@ -3586,17 +3586,17 @@ mlir::Value CodeGenTileLangNPUIRDEV::VisitExpr_(const CallNode *op) {
   } else if (op->op.same_as(Op::Get("tl.npuir_ln"))) {
     UnaryVecOpCodegen<tvm::tl::NpuirLn, mlir::linalg::UnaryFn::log>(op);
   } else if (op->op.same_as(Op::Get("tl.npuir_relu"))) {
-    UnaryVecOpCodegen<tvm::tl::NpuirRelu, hfusion::UnaryFn::relu>(op);
+    UnaryVecOpCodegen<tvm::tl::NpuirRelu, mlir::hfusion::UnaryFn::relu>(op);
   } else if (op->op.same_as(Op::Get("tl.npuir_sqrt"))) {
     UnaryVecOpCodegen<tvm::tl::NpuirSqrt, mlir::linalg::UnaryFn::sqrt>(op);
   } else if (op->op.same_as(Op::Get("tl.npuir_rsqrt"))) {
-    UnaryVecOpCodegen<tvm::tl::NpuirRsqrt, mlir::linalg::UnaryFn::rsqrt>(op);
+    UnaryVecOpCodegen<tvm::tl::NpuirRsqrt, mlir::hfusion::UnaryFn::rsqrt>(op);
   } else if (op->op.same_as(Op::Get("tl.npuir_abs"))) {
     UnaryVecOpCodegen<tvm::tl::NpuirAbs, mlir::linalg::UnaryFn::abs>(op);
   } else if (op->op.same_as(Op::Get("tl.npuir_rec"))) {
     UnaryVecOpCodegen<tvm::tl::NpuirRec, mlir::linalg::UnaryFn::reciprocal>(op);
   } else if (op->op.same_as(Op::Get("tl.npuir_not"))) {
-    UnaryVecOpCodegen<tvm::tl::NpuirNot, hfusion::UnaryFn::vnot>(op);
+    UnaryVecOpCodegen<tvm::tl::NpuirNot, mlir::hfusion::UnaryFn::vnot>(op);
   } else if (op->op.same_as(Op::Get("tl.npuir_select"))) {
     VselectCodegen(op);
   } else if (op->op.same_as(Op::Get("tl.npuir_cmp"))) {
diff --git a/testing/npuir/linalg_ops/test_rsqrt_dev.py b/testing/npuir/hfusion_ops/test_rsqrt_dev.py
similarity index 100%
rename from testing/npuir/linalg_ops/test_rsqrt_dev.py
rename to testing/npuir/hfusion_ops/test_rsqrt_dev.py