Add MLIR conv3d support (#3138)

ROCm · Jun 7, 2024 · fd9af84 · fd9af84
1 parent 2690ecf
commit fd9af84
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/targets/gpu/fuse_mlir.cpp b/src/targets/gpu/fuse_mlir.cpp
@@ -257,8 +257,14 @@ auto is_mlir_conv(mlir_mode mode)
         value v    = ins->get_operator().to_value();
         auto group = v.at("group").to<int>();
         // Avoid MLIR assertion: Index < Length && "Invalid index!"
+#ifdef _WIN32
+        // Temporarily make it available only on Windows
+        if(ins->get_shape().lens().size() != 4 and group > 1)
+            return false;
+#else
         if(ins->get_shape().lens().size() != 4)
             return false;
+#endif
         if(contains({shape::fp8e4m3fnuz_type, shape::int8_type}, input.type()))
             return true;
         if(mode == mlir_mode::all)