.

tenstorrent · Feb 7, 2025 · 3a67d6d · 3a67d6d
1 parent 3004131
commit 3a67d6d
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 9 deletions.
diff --git a/runtime/include/tt/runtime/detail/workarounds.h b/runtime/include/tt/runtime/detail/workarounds.h
@@ -23,7 +23,7 @@ struct Env {
       ;
 #else
   {
-    return Env(true, true, true, true, true);
+    return Env(true, true, true, true);
   }
 #endif
   // TODO(bug #1124): We're currently swapping the operands for binary ops

diff --git a/runtime/tools/python/ttrt/common/run.py b/runtime/tools/python/ttrt/common/run.py
@@ -119,13 +119,6 @@ def initialize_api():
             choices=[True, False],
             help="enable async mode device execution for TTNN runtime",
         )
-        Run.register_arg(
-            name="--disable-maxpool2d-preshard",
-            type=bool,
-            default=False,
-            choices=[True, False],
-            help="disable maxpool2d preshard workaround",
-        )
         Run.register_arg(
             name="--disable-swap-binary-operands",
             type=bool,
@@ -421,7 +414,6 @@ def convert_input_layouts(device, inputs, fbb, program_index):
             debug_env = ttrt.runtime.DebugEnv.get(self["--load-kernels-from-disk"])
             self.logging.debug(f"setting tt runtime debug env={debug_env}")
             workaround_env = ttrt.runtime.WorkaroundEnv.get(
-                not self["--disable-maxpool2d-preshard"],
                 not self["--disable-swap-binary-operands"],
                 not self["--disable-read-update-index-for-kv-cache"],
                 not self["--disable-to-layout-api-assume-single-chip"],
-Original file line number
+Diff line change
@@ Expand Up / @@ -23,7 +23,7 @@ struct Env { @@
           ;
     #else
       {
-        return Env(true, true, true, true, true);
+        return Env(true, true, true, true);
       }
     #endif
       // TODO(bug #1124): We're currently swapping the operands for binary ops
@@ Expand Down @@