Add ttforge-sweep in github workflow.

Signed-off-by: Nilaykumar Patel <[email protected]>
tenstorrent · Jan 9, 2025 · e1f8727 · e1f8727
1 parent 76c3d64
commit e1f8727
Show file tree

Hide file tree

Showing 4 changed files with 7,235 additions and 7,190 deletions.
diff --git a/.github/workflows/ttnn-run-sweeps.yaml b/.github/workflows/ttnn-run-sweeps.yaml
@@ -388,6 +388,7 @@ on:
           - conv2d.full.conv2d_sharding
           - conv2d.full.conv2d_sliding_window
           - conv2d.short.conv2d_short_sweep
+          - conv2d.short.conv2d_ttforge_sweep
           - pooling.global_avg_pool2d
           - pooling.max_pool2d
           - max_pool2d.short.max_pool2d_short_sweep

diff --git a/tests/sweep_framework/sweep_utils/conv2d_common.py b/tests/sweep_framework/sweep_utils/conv2d_common.py
@@ -259,9 +259,7 @@ def run_conv2d_short_sweep(
     if is_forge_suite:
         input_layout = ttnn.Layout(input_layout)
         input_dtype = ttnn.DataType(input_dtype)
-        input_memory_config = (
-            ttnn.DRAM_MEMORY_CONFIG if input_buffer_type == int(ttnn.BufferType.DRAM) else ttnn.L1_MEMORY_CONFIG
-        )
+        input_memory_config = ttnn.DRAM_MEMORY_CONFIG if input_buffer_type == "dram" else ttnn.L1_MEMORY_CONFIG
         tt_input_tensor = ttnn.from_torch(
             torch_input_tensor, dtype=input_dtype, layout=input_layout, device=device, memory_config=input_memory_config
         )