#0: Bump tg/tgg compile times for Resnet

tenstorrent · Oct 16, 2024 · c137d33 · c137d33
1 parent ef06fc4
commit c137d33
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 8 deletions.
diff --git a/models/demos/tg/resnet50/tests/test_perf_e2e_resnet50.py b/models/demos/tg/resnet50/tests/test_perf_e2e_resnet50.py
@@ -13,7 +13,7 @@
 @pytest.mark.parametrize("device_params", [{"l1_small_size": 24576}], indirect=True)
 @pytest.mark.parametrize(
     "device_batch_size, enable_async_mode, expected_inference_time, expected_compile_time",
-    ((16, True, 0.0370, 60),),
+    ((16, True, 0.0370, 120),),
     indirect=["enable_async_mode"],
 )
 @pytest.mark.parametrize(
@@ -48,7 +48,7 @@ def test_perf(
 @pytest.mark.parametrize("device_params", [{"l1_small_size": 32768, "trace_region_size": 1500000}], indirect=True)
 @pytest.mark.parametrize(
     "device_batch_size, enable_async_mode, expected_inference_time, expected_compile_time",
-    ((16, True, 0.0081, 60),),
+    ((16, True, 0.0081, 120),),
     indirect=["enable_async_mode"],
 )
 @pytest.mark.parametrize(
@@ -83,7 +83,7 @@ def test_perf_trace(
 @pytest.mark.parametrize("device_params", [{"l1_small_size": 32768, "num_command_queues": 2}], indirect=True)
 @pytest.mark.parametrize(
     "device_batch_size, enable_async_mode, expected_inference_time, expected_compile_time",
-    ((16, True, 0.0391, 60),),
+    ((16, True, 0.0391, 120),),
     indirect=["enable_async_mode"],
 )
 def test_perf_2cqs(
@@ -115,7 +115,7 @@ def test_perf_2cqs(
 )
 @pytest.mark.parametrize(
     "device_batch_size, enable_async_mode, expected_inference_time, expected_compile_time",
-    ((16, True, 0.0069, 60),),
+    ((16, True, 0.0069, 120),),
     indirect=["enable_async_mode"],
 )
 def test_perf_trace_2cqs(

diff --git a/models/demos/tgg/resnet50/tests/test_perf_e2e_resnet50.py b/models/demos/tgg/resnet50/tests/test_perf_e2e_resnet50.py
@@ -13,7 +13,7 @@
 @pytest.mark.parametrize("device_params", [{"l1_small_size": 24576}], indirect=True)
 @pytest.mark.parametrize(
     "device_batch_size, enable_async_mode, expected_inference_time, expected_compile_time",
-    ((16, True, 0.0700, 60),),
+    ((16, True, 0.0830, 180),),
     indirect=["enable_async_mode"],
 )
 @pytest.mark.parametrize(
@@ -48,7 +48,7 @@ def test_perf(
 @pytest.mark.parametrize("device_params", [{"l1_small_size": 32768, "trace_region_size": 1500000}], indirect=True)
 @pytest.mark.parametrize(
     "device_batch_size, enable_async_mode, expected_inference_time, expected_compile_time",
-    ((16, True, 0.0081, 60),),
+    ((16, True, 0.0081, 180),),
     indirect=["enable_async_mode"],
 )
 @pytest.mark.parametrize(
@@ -83,7 +83,7 @@ def test_perf_trace(
 @pytest.mark.parametrize("device_params", [{"l1_small_size": 32768, "num_command_queues": 2}], indirect=True)
 @pytest.mark.parametrize(
     "device_batch_size, enable_async_mode, expected_inference_time, expected_compile_time",
-    ((16, True, 0.0730, 60),),
+    ((16, True, 0.0830, 180),),
     indirect=["enable_async_mode"],
 )
 @pytest.mark.parametrize(
@@ -120,7 +120,7 @@ def test_perf_2cqs(
 )
 @pytest.mark.parametrize(
     "device_batch_size, enable_async_mode, expected_inference_time, expected_compile_time",
-    ((16, True, 0.0073, 60),),
+    ((16, True, 0.0073, 180),),
     indirect=["enable_async_mode"],
 )
 @pytest.mark.parametrize(