[Fix CI] Fix TT-Explorer tests in Ci. (#1283)

tenstorrent · Nov 15, 2024 · 16edb11 · 16edb11
1 parent 85b17d8
commit 16edb11
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tools/explorer/test/run_tests.py b/tools/explorer/test/run_tests.py
@@ -13,11 +13,11 @@
 PORT = 8002
 COMMAND_URL = "http://" + HOST + ":" + str(PORT) + "/apipost/v1/send_command"
 TEST_LOAD_MODEL_PATHS = [
-    "test/ttmlir/Dialect/TTNN/mnist_sharding.mlir",
+    "test/ttmlir/Dialect/TTNN/optimizer/mnist_sharding.mlir",
     "tools/explorer/test/models/*.mlir",
 ]
 TEST_EXECUTE_MODEL_PATHS = [
-    "test/ttmlir/Silicon/TTNN/sharded/mnist_sharding_tiled.mlir",
+    "test/ttmlir/Silicon/TTNN/optimizer/mnist_sharding_tiled.mlir",
 ]
 
 
@@ -97,14 +97,14 @@ def test_execute_model(model_path):
 
 def test_execute_mnist_l1_interleaved():
     execute_command(
-        "test/ttmlir/Silicon/TTNN/sharded/mnist_sharding_tiled.mlir",
+        "test/ttmlir/Silicon/TTNN/optimizer/mnist_sharding_tiled.mlir",
         {"optimizationPolicy": "L1 Interleaved"},
     )
 
 
 def test_execute_mnist_optimizer_disabled():
     execute_command(
-        "test/ttmlir/Silicon/TTNN/sharded/mnist_sharding_tiled.mlir",
+        "test/ttmlir/Silicon/TTNN/optimizer/mnist_sharding_tiled.mlir",
         {"optimizationPolicy": "Optimizer Disabled"},
     )