Don't hoist bit extend

Signed-off-by: Ian Wood <ianwood2024@u.northwestern.edu>
iree-org · Jan 30, 2025 · dd66e55 · dd66e55
1 parent cebecc1
commit dd66e55
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/compiler/src/iree/compiler/ExternalInterfaces/UtilExternalModels.cpp b/compiler/src/iree/compiler/ExternalInterfaces/UtilExternalModels.cpp
@@ -12,6 +12,7 @@
 #include "iree/compiler/Dialect/Encoding/IR/EncodingOps.h"
 #include "iree/compiler/Dialect/LinalgExt/IR/LinalgExtDialect.h"
 #include "iree/compiler/Dialect/LinalgExt/IR/LinalgExtOps.h"
+#include "iree/compiler/Dialect/LinalgExt/Utils/Utils.h"
 #include "iree/compiler/Dialect/Util/IR/UtilDialect.h"
 #include "iree/compiler/Dialect/Util/IR/UtilOps.h"
 #include "iree/compiler/Dialect/Util/IR/UtilTypes.h"
@@ -340,6 +341,9 @@ struct HoistableLinalgOpInterface
     auto genericOp = llvm::dyn_cast<linalg::GenericOp>(op);
     if (!genericOp)
       return !isa<linalg::FillOp>(op);
+    if (IREE::LinalgExt::isBitExtendOp(op)) {
+      return false;
+    }
     // Generally, we prefer to not hoist broadcasts.
     // Detect op that only broadcast input as fusing them makes the new
     // op cheaper.