feat(nvidia): add dali (#21646)

hongbo-miao · Dec 16, 2024 · 8890303 · 8890303
1 parent 8ba95fd
commit 8890303
Show file tree

Hide file tree

Showing 12 changed files with 777 additions and 73 deletions.
diff --git a/.github/workflows/.lint.yml b/.github/workflows/.lint.yml
@@ -246,7 +246,7 @@ jobs:
       - name: Install uv
         uses: astral-sh/setup-uv@v4.2.0
         with:
-          version: 0.5.6
+          version: 0.5.9
           enable-cache: true
           cache-dependency-glob: uv.lock
       - name: Set up Python
@@ -329,7 +329,7 @@ jobs:
       - name: Install uv
         uses: astral-sh/setup-uv@v4.2.0
         with:
-          version: 0.5.6
+          version: 0.5.9
           enable-cache: true
           cache-dependency-glob: uv.lock
       - name: Set up Python
@@ -649,7 +649,7 @@ jobs:
       - name: Install uv
         uses: astral-sh/setup-uv@v4.2.0
         with:
-          version: 0.5.6
+          version: 0.5.9
           enable-cache: true
           cache-dependency-glob: uv.lock
       - name: Set up Python
@@ -696,7 +696,7 @@ jobs:
       - name: Install uv
         uses: astral-sh/setup-uv@v4.2.0
         with:
-          version: 0.5.6
+          version: 0.5.9
           enable-cache: true
           cache-dependency-glob: uv.lock
       - name: Set up Python
@@ -1208,7 +1208,7 @@ jobs:
       - name: Install uv
         uses: astral-sh/setup-uv@v4.2.0
         with:
-          version: 0.5.6
+          version: 0.5.9
           enable-cache: true
           cache-dependency-glob: uv.lock
       - name: Set up Python
@@ -1286,7 +1286,7 @@ jobs:
       - name: Install uv
         uses: astral-sh/setup-uv@v4.2.0
         with:
-          version: 0.5.6
+          version: 0.5.9
           enable-cache: true
           cache-dependency-glob: uv.lock
       - name: Set up Python
@@ -1336,7 +1336,7 @@ jobs:
       - name: Install uv
         uses: astral-sh/setup-uv@v4.2.0
         with:
-          version: 0.5.6
+          version: 0.5.9
           enable-cache: true
           cache-dependency-glob: uv.lock
       - name: Set up Python

diff --git a/.github/workflows/.static-type-check.yml b/.github/workflows/.static-type-check.yml
@@ -147,6 +147,7 @@ jobs:
           uv run poe static-type-check-python --package=hm-locust
           uv run poe static-type-check-python --package=hm-xxhash
           uv run poe static-type-check-python --package=machine-learning.convolutional-neural-network
+          uv run poe static-type-check-python --package=machine-learning.dali
           uv run poe static-type-check-python --package=machine-learning.feature-store
           uv run poe static-type-check-python --package=machine-learning.graph-neural-network
           uv run poe static-type-check-python --package=machine-learning.hm-gradio.applications.classify-image