Keep MIGRAPHX_ENABLE_EXTRA_MLIR

fdd29d75 · Paul · 6f3784aa · fdd29d75
Commit fdd29d75 authored Nov 16, 2023 by Paul
Show whitespace changes
Inline Side-by-side

Showing with 3 additions and 0 deletions

src/targets/gpu/fuse_mlir.cpp src/targets/gpu/fuse_mlir.cpp +3 -0

No files found.
--- a/src/targets/gpu/fuse_mlir.cpp
+++ b/src/targets/gpu/fuse_mlir.cpp
@@ -36,6 +36,7 @@ struct module;
 namespace gpu {
+MIGRAPHX_DECLARE_ENV_VAR(MIGRAPHX_ENABLE_EXTRA_MLIR);
 MIGRAPHX_DECLARE_ENV_VAR(MIGRAPHX_DISABLE_MLIR);
 bool mlir_enabled()
@@ -171,6 +172,8 @@ auto is_mlir_dot(mlir_mode mode)
        float g = a.elements() / (m * k);
        if(k > 1024)
            return false;
+        if(enabled(MIGRAPHX_ENABLE_EXTRA_MLIR{}))
+            return true;
        auto ratio = std::sqrt(g) * m * n / k;
        if(ratio < 2048)
            return false;