#0: Merge branch 'jvega/tiny_tiles_all_gather' of github.com:tenstorr…

…ent/tt-metal into jvega/tiny_tiles_all_gather
tenstorrent · Nov 8, 2024 · 9221c66 · 9221c66
2 parents dc21b52 + 4a90c9f
commit 9221c66
Show file tree

Hide file tree

Showing 106 changed files with 2,999 additions and 1,296 deletions.
diff --git a/.github/workflows/pipeline-select-t3k.yaml b/.github/workflows/pipeline-select-t3k.yaml
@@ -12,6 +12,10 @@ on:
           - RelWithDebInfo
           - CI
         default: "Release"
+      extra-tag:
+        required: true
+        type: string
+        default: "in-service"
       build-with-tracy:
         required: false
         type: boolean
@@ -54,29 +58,41 @@ jobs:
     needs: build-artifact
     secrets: inherit
     uses: ./.github/workflows/t3000-unit-tests-impl.yaml
+    with:
+      extra-tag: ${{ inputs.extra-tag }}
     if: ${{ inputs.t3000-unit }}
   t3000-demo-tests:
     needs: build-artifact
     secrets: inherit
     uses: ./.github/workflows/t3000-demo-tests-impl.yaml
+    with:
+      extra-tag: ${{ inputs.extra-tag }}
     if: ${{ inputs.t3000-demo }}
   t3000-frequent-tests:
     needs: build-artifact
     secrets: inherit
     uses: ./.github/workflows/t3000-frequent-tests-impl.yaml
+    with:
+      extra-tag: ${{ inputs.extra-tag }}
     if: ${{ inputs.t3000-frequent }}
   t3000-nightly-tests:
     needs: build-artifact
     secrets: inherit
     uses: ./.github/workflows/t3000-nightly-tests-impl.yaml
+    with:
+      extra-tag: ${{ inputs.extra-tag }}
     if: ${{ inputs.t3000-nightly }}
   t3000-model-perf-tests:
     needs: build-artifact
     secrets: inherit
     uses: ./.github/workflows/t3000-model-perf-tests-impl.yaml
+    with:
+      extra-tag: ${{ inputs.extra-tag }}
     if: ${{ inputs.t3000-model-perf }}
   t3000-profiler-tests:
     needs: build-artifact
     secrets: inherit
     uses: ./.github/workflows/t3000-profiler-tests-impl.yaml
+    with:
+      extra-tag: ${{ inputs.extra-tag }}
     if: ${{ inputs.t3000-profiler }}
diff --git a/.github/workflows/t3000-demo-tests-impl.yaml b/.github/workflows/t3000-demo-tests-impl.yaml
@@ -2,6 +2,11 @@ name: "[internal] T3000 demo tests impl"
 
 on:
   workflow_call:
+    inputs:
+      extra-tag:
+        required: false
+        type: string
+        default: "in-service"
 
 jobs:
   t3000-demo-tests:
@@ -23,7 +28,11 @@ jobs:
       ARCH_NAME: ${{ matrix.test-group.arch }}
       LOGURU_LEVEL: INFO
       LD_LIBRARY_PATH: ${{ github.workspace }}/build/lib
-    runs-on: ["arch-wormhole_b0", "config-t3000", "in-service", "pipeline-perf"]
+    runs-on:
+      - arch-wormhole_b0
+      - config-t3000
+      - pipeline-perf
+      - ${{ inputs.extra-tag }}
     steps:
       - uses: tenstorrent-metal/metal-workflows/.github/actions/checkout-with-submodule-lfs@v2.0.0
       - name: Enable performance mode

diff --git a/.github/workflows/t3000-frequent-tests-impl.yaml b/.github/workflows/t3000-frequent-tests-impl.yaml
@@ -2,6 +2,11 @@ name: "[internal] T3000 frequent tests impl"
 
 on:
   workflow_call:
+    inputs:
+      extra-tag:
+        required: false
+        type: string
+        default: "in-service"
 
 jobs:
   t3000-frequent-tests:
@@ -27,7 +32,11 @@ jobs:
       ARCH_NAME: ${{ matrix.test-group.arch }}
       LOGURU_LEVEL: INFO
       LD_LIBRARY_PATH: ${{ github.workspace }}/build/lib
-    runs-on: ["arch-wormhole_b0", "config-t3000", "in-service", "pipeline-functional"]
+    runs-on:
+      - arch-wormhole_b0
+      - config-t3000
+      - pipeline-functional
+      - ${{ inputs.extra-tag }}
     steps:
       - uses: tenstorrent-metal/metal-workflows/.github/actions/checkout-with-submodule-lfs@v2.0.0
       - uses: ./.github/actions/ensure-active-weka-mount

diff --git a/.github/workflows/t3000-model-perf-tests-impl.yaml b/.github/workflows/t3000-model-perf-tests-impl.yaml
@@ -2,6 +2,11 @@ name: "[internal] T3000 model perf tests impl"
 
 on:
   workflow_call:
+    inputs:
+      extra-tag:
+        required: false
+        type: string
+        default: "in-service"
 
 jobs:
 
@@ -25,7 +30,11 @@ jobs:
       ARCH_NAME: ${{ matrix.test-group.arch }}
       LOGURU_LEVEL: INFO
       LD_LIBRARY_PATH: ${{ github.workspace }}/build/lib
-    runs-on: ["arch-wormhole_b0", "config-t3000", "in-service", "pipeline-perf"]
+    runs-on:
+      - arch-wormhole_b0
+      - config-t3000
+      - pipeline-perf
+      - ${{ inputs.extra-tag }}
     steps:
       - uses: tenstorrent-metal/metal-workflows/.github/actions/checkout-with-submodule-lfs@v2.0.0
       - name: Enable performance mode

diff --git a/.github/workflows/t3000-nightly-tests-impl.yaml b/.github/workflows/t3000-nightly-tests-impl.yaml
@@ -2,6 +2,11 @@ name: "[internal] T3000 nightly tests impl"
 
 on:
   workflow_call:
+    inputs:
+      extra-tag:
+        required: false
+        type: string
+        default: "in-service"
 
 jobs:
   t3000-nightly-tests:
@@ -18,7 +23,11 @@ jobs:
       ARCH_NAME: ${{ matrix.test-group.arch }}
       LOGURU_LEVEL: INFO
       LD_LIBRARY_PATH: ${{ github.workspace }}/build/lib
-    runs-on: ["arch-wormhole_b0", "config-t3000", "in-service", "pipeline-functional"]
+    runs-on:
+      - arch-wormhole_b0
+      - config-t3000
+      - pipeline-functional
+      - ${{ inputs.extra-tag }}
     steps:
       - uses: tenstorrent-metal/metal-workflows/.github/actions/checkout-with-submodule-lfs@v2.0.0
       - name: Set up dynamic env vars for build

diff --git a/.github/workflows/t3000-profiler-tests-impl.yaml b/.github/workflows/t3000-profiler-tests-impl.yaml
@@ -2,6 +2,11 @@ name: "[internal] T3000 profiler tests impl"
 
 on:
   workflow_call:
+    inputs:
+      extra-tag:
+        required: false
+        type: string
+        default: "in-service"
 
 jobs:
   t3000-profiler-tests:
@@ -12,7 +17,7 @@ jobs:
           {
             name: "T3000 profiler tests",
             arch: wormhole_b0,
-            runs-on: ["arch-wormhole_b0", "config-t3000", "in-service", "pipeline-perf"],
+            runs-on: ["arch-wormhole_b0", "config-t3000", "pipeline-perf", "${{ inputs.extra-tag }}"],
             cmd: './tests/scripts/run_profiler_regressions.sh'
           },
         ]

diff --git a/.github/workflows/t3000-unit-tests-impl.yaml b/.github/workflows/t3000-unit-tests-impl.yaml
@@ -2,6 +2,11 @@ name: "[internal] T3000 unit tests impl"
 
 on:
   workflow_call:
+    inputs:
+      extra-tag:
+        required: false
+        type: string
+        default: "in-service"
 
 jobs:
   t3000-unit-tests:
@@ -28,7 +33,11 @@ jobs:
       ARCH_NAME: ${{ matrix.test-group.arch }}
       LOGURU_LEVEL: INFO
       LD_LIBRARY_PATH: ${{ github.workspace }}/build/lib
-    runs-on: ["arch-wormhole_b0", "config-t3000", "in-service", "pipeline-functional"]
+    runs-on:
+      - arch-wormhole_b0
+      - config-t3000
+      - pipeline-functional
+      - ${{ inputs.extra-tag }}
     steps:
       - uses: tenstorrent-metal/metal-workflows/.github/actions/checkout-with-submodule-lfs@v2.0.0
       - name: Set up dynamic env vars for build

diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -263,15 +263,13 @@ if("$ENV{ARCH_NAME}" STREQUAL "wormhole_b0")
             tt_metal/hw/inc/wormhole
             ${PROJECT_SOURCE_DIR}/tt_metal/hw/inc/wormhole/wormhole_b0_defines
             ${UMD_HOME}/device/wormhole
-            ${UMD_HOME}/src/firmware/riscv/wormhole
     )
 else()
     target_include_directories(
         metal_header_directories
         INTERFACE
             ${PROJECT_SOURCE_DIR}/tt_metal/hw/inc/$ENV{ARCH_NAME}
             ${UMD_HOME}/device/$ENV{ARCH_NAME}
-            ${UMD_HOME}/src/firmware/riscv/$ENV{ARCH_NAME}
     )
 endif()