Change default value for ONEFLOW_MLIR_PREFER_NHWC (#10410)

Oneflow-Inc · Jan 20, 2024 · 9afbc9b · 9afbc9b
1 parent 2e331e1
commit 9afbc9b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/python/oneflow/framework/infer_compiler/with_oneflow_compile.py b/python/oneflow/framework/infer_compiler/with_oneflow_compile.py
@@ -388,7 +388,8 @@ def __init__(self, model):
         os.environ.setdefault("ONEFLOW_MLIR_FUSE_FORWARD_OPS", "1")
         os.environ.setdefault("ONEFLOW_MLIR_FUSE_OPS_WITH_BACKWARD_IMPL", "1")
         os.environ.setdefault("ONEFLOW_MLIR_GROUP_MATMUL", "1")
-        os.environ.setdefault("ONEFLOW_MLIR_PREFER_NHWC", "1")
+        # TODO(lml): enable ONEFLOW_MLIR_PREFER_NHWC when related bug fix.
+        os.environ.setdefault("ONEFLOW_MLIR_PREFER_NHWC", "0")
         os.environ.setdefault("ONEFLOW_KERNEL_ENABLE_FUSED_CONV_BIAS", "1")
         os.environ.setdefault("ONEFLOW_KERNEL_ENABLE_FUSED_LINEAR", "1")
         os.environ.setdefault(