[WebGPU] Add gating logic for subgroup shuffle primitives

.asf.yaml

@@ -16,7 +16,7 @@
 # under the License.
 
 github:
-  description: "Open deep learning compiler stack for cpu, gpu and specialized accelerators"
+  description: "Open Machine Learning Compiler Framework"
   homepage: https://tvm.apache.org/
   labels:
     - tvm
@@ -33,6 +33,12 @@ github:
     - spirv
     - machine-learning
 
+  features:
+    # Enable issue management
+    issues: true
+    # Enable projects for project management boards
+    projects: true
+
   # Triage perm for collaborators(test run)
   #
   # The perm is given based on needs and not based on
@@ -45,26 +51,39 @@ github:
   # participation, permission is given on a three month
   # cycle. PMC may review and recycle slots when necessary.
   collaborators:
-    - hpanda-naut
-    - denise-k
-    - janetsc
-    - naut-thomas
     - tvm-bot  # For automated feedback in PR review.
 
   # See https://cwiki.apache.org/confluence/display/INFRA/Git+-+.asf.yaml+features#Git.asf.yamlfeatures-Branchprotection
   protected_branches:
     main:
       required_status_checks:
         contexts:
-          - unity/pr-head
           - arm/pr-head
           - cpu/pr-head
           - docker/pr-head
           - gpu/pr-head
-          - hexagon/pr-head
-          - i386/pr-head
-          - lint/pr-head
           - wasm/pr-head
 
       required_pull_request_reviews:
         required_approving_review_count: 1
+
+  enabled_merge_buttons:
+    # enable squash button:
+    squash: true
+    # default commit message when merging with a squash commit
+    # can either be: DEFAULT | PR_TITLE | PR_TITLE_AND_COMMIT_DETAILS | PR_TITLE_AND_DESC
+    squash_commit_message: PR_TITLE_AND_DESC
+    # enable merge button:
+    merge: false
+    # default commit message when merging with a merge commit
+    # can either be: DEFAULT | PR_TITLE | PR_TITLE_AND_DESC
+    merge_commit_message: DEFAULT
+    # enable rebase button for rare use.
+    rebase: true
+
+notifications:
+  commits: commits@tvm.apache.org
+  issues: discuss-archive@tvm.apache.org
+  pullrequests: discuss-archive@tvm.apache.org
+  jobs: discuss-archive@tvm.apache.org
+  discussions: discuss-archive@tvm.apache.org

.clang-format

@@ -4,5 +4,5 @@
 # https://clang.llvm.org/docs/ClangFormat.html
 BasedOnStyle: Google
 DerivePointerAlignment: false
-ColumnLimit:     100
+ColumnLimit: 100
 PointerAlignment: Left

.github/CODEOWNERSHIP

@@ -15,7 +15,7 @@
 # specific language governing permissions and limitations
 # under the License.
 
-# Github code owners file
+# GitHub code owners file
 # This file is used as a convenient tool to map
 # committers' areas of expertise and faciliate the review process.
 #
@@ -66,7 +66,7 @@ web/** @tqchen @jroesch
 docker/** @areusch @leandron @jroesch
 
 # Conda
-conda/** @tqchen @junrushao1994 @comaniac
+tests/conda/** @tqchen @junrushao1994 @comaniac
 
 # CMake
 cmake/** @jroesch @tqchen @areusch @junrushao1994 @comaniac

.github/actions/setup/action.yml

@@ -1,39 +1,42 @@
 runs:
  using: "composite"
  steps:
-  - uses: actions/cache@v3
+  - uses: actions/cache@cdf6c1fa76f9f475f3d7449005a359c84ca0f306 # v5.0.3
     env:
-      CACHE_NUMBER: 1
+      CACHE_NUMBER: 2
     with:
       path: ~/conda_pkgs_dir
-      key: ${{ runner.os }}-conda-${{ env.CACHE_NUMBER }}-${{ hashFiles('conda/build-environment.yaml') }}
-  - uses: conda-incubator/setup-miniconda@v3
+      key: ${{ runner.os }}-conda-${{ env.CACHE_NUMBER }}-${{ hashFiles('tests/conda/build-environment.yaml') }}
+  - uses: conda-incubator/setup-miniconda@fc2d68f6413eb2d87b895e92f8584b5b94a10167 # v3.3.0
     continue-on-error: true
     id: conda1
     with:
       activate-environment: tvm-build
       channel-priority: strict
-      environment-file: conda/build-environment.yaml
+      environment-file: tests/conda/build-environment.yaml
       auto-activate-base: false
-      conda-solver: classic
-      use-only-tar-bz2: true
-      python-version: 3.9
-      condarc-file: conda/condarc
-  - uses: conda-incubator/setup-miniconda@v3
+      miniforge-version: latest
+      python-version: "3.10"
+      condarc-file: tests/conda/condarc
+  - uses: conda-incubator/setup-miniconda@fc2d68f6413eb2d87b895e92f8584b5b94a10167 # v3.3.0
     if: steps.conda1.outcome == 'failure'
     with:
       activate-environment: tvm-build
       channel-priority: strict
-      environment-file: conda/build-environment.yaml
+      environment-file: tests/conda/build-environment.yaml
       auto-activate-base: false
-      conda-solver: classic
+      miniforge-version: latest
       use-only-tar-bz2: true
-      python-version: 3.9
-      condarc-file: conda/condarc
+      python-version: "3.10"
+      condarc-file: tests/conda/condarc
   - name: Conda info
     shell: pwsh
     run: |
-      conda info
-      conda list
-      conda info --envs
-      conda list --name base
+      mamba info
+      mamba list
+      mamba info --envs
+      mamba list --name base
+  - name: Install tvm-ffi pip package
+    shell: bash -l {0}
+    run: |
+      pip install -v ./3rdparty/tvm-ffi

.github/workflows/cc_bot.yml

@@ -35,7 +35,7 @@ jobs:
     if: github.repository == 'apache/tvm'
     runs-on: ubuntu-latest
     steps:
-      - uses: actions/checkout@v2
+      - uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd # v6.0.2
         with:
           submodules: "recursive"
       - name: Add cc'ed reviewers

.github/workflows/lint.yml

@@ -0,0 +1,42 @@
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements.  See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership.  The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License.  You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
+name: Lint
+
+on:
+  push:
+    branches: [main]
+  pull_request:
+    branches: [main]
+
+concurrency:
+  group: Lint-${{ github.event.pull_request.number || github.sha }}
+  cancel-in-progress: true
+
+jobs:
+  lint:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
+        with:
+          fetch-depth: 0
+          fetch-tags: true
+      - name: Set up uv
+        uses: astral-sh/setup-uv@b75a909f75acd358c2196fb9a5f1299a9a8868a4  # v6.7.0
+      - name: Set up Python environment
+        run: uv sync --group lint --no-install-project
+      - uses: pre-commit/action@2c7b3805fd2a0fd8c1884dcaebf91fc102a13ecd  # v3.0.1

.github/workflows/main.yml

@@ -39,32 +39,26 @@ jobs:
     if: ${{ github.repository == 'apache/tvm' }}
     runs-on: macOS-latest
     steps:
-      - uses: actions/checkout@v2
+      - uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd # v6.0.2
         with:
           submodules: 'recursive'
       - name: Set up environment
         uses: ./.github/actions/setup
-      - name: Conda Build
+      - name: Install LLVM dependencies
+        shell: bash -l {0}
+        run: |
+          conda install -c conda-forge llvmdev cmake ninja zlib
+      - name: Build TVM wheel
+        shell: bash -l {0}
+        run: |
+          pip install scikit-build-core
+          export CMAKE_ARGS="-DUSE_LLVM=ON -DBUILD_TESTING=OFF"
+          pip wheel --no-deps -w dist . -v
+      - name: Install TVM from wheel
         shell: bash -l {0}
-        run: >-
-          conda build --output-folder=conda/pkg  conda/recipe &&
-          conda install tvm -c ./conda/pkg
-      - name: Build iOS RPC
         run: |
-          IOS_VERSION="14.0"
-          CMAKE_FLAGS="-DCMAKE_BUILD_TYPE=Release \
-                       -DCMAKE_SYSTEM_NAME=iOS \
-                       -DCMAKE_SYSTEM_VERSION=${IOS_VERSION} \
-                       -DCMAKE_OSX_SYSROOT=iphonesimulator \
-                       -DCMAKE_OSX_ARCHITECTURES=x86_64 \
-                       -DCMAKE_OSX_DEPLOYMENT_TARGET=14.0 \
-                       -DCMAKE_BUILD_WITH_INSTALL_NAME_DIR=ON \
-                       -DUSE_IOS_RPC=ON"
+          pip install dist/*.whl
 
-          mkdir build-ios-simulator
-          cd build-ios-simulator
-          cmake .. ${CMAKE_FLAGS}
-          cmake --build . --target ios_rpc
       - name: Test
         shell: bash -l {0}
         run: >-
@@ -80,102 +74,31 @@ jobs:
           python -m pytest -v -s 'tests/python/codegen/test_target_codegen_metal.py'
           python -m pytest -v -s 'tests/python/codegen/test_target_codegen_gpu_common.py'
           python -m pytest -v -s 'tests/python/codegen/test_gpu_codegen_allreduce.py::test_allreduce_sum[dims0-metal]'
-#      - name: Test iOS RPC
-#        shell: bash -l {0}
-#        run: >-
-#          python -m pip install tornado psutil cloudpickle &&
-#          export PYTHONPATH=tests/python/contrib:${PYTHONPATH} &&
-#          export BUNDLE_ID=org.apache.tvmrpc &&
-#          export BUNDLE_PATH=build-ios-simulator/apps/ios_rpc/ios_rpc/src/ios_rpc-build/Release-iphonesimulator/tvmrpc.app &&
-#          python -m pytest -v tests/python/contrib/test_rpc_server_device.py
 
   Windows:
     if: ${{ github.repository == 'apache/tvm' }}
-    runs-on: windows-2019
+    runs-on: windows-latest
     steps:
-      - uses: actions/checkout@v2
+      - uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd # v6.0.2
         with:
           submodules: 'recursive'
       - name: Set up environment
         uses: ./.github/actions/setup
-      - name: Conda Build
+      - name: Install LLVM dependencies
         shell: cmd /C call {0}
-        run: >-
-          conda build --output-folder=conda/pkg conda/recipe &&
-          conda install tvm -c ./conda/pkg
+        run: |
+          conda install -c conda-forge llvmdev cmake ninja zlib libxml2-devel
+      - name: Install TVM
+        shell: cmd /C call {0}
+        run: |
+          pip install scikit-build-core
+          set CMAKE_ARGS=-DUSE_LLVM=ON -DBUILD_TESTING=OFF
+          pip install --no-deps . -v
+      - name: Install test dependencies
+        shell: cmd /C call {0}
+        run: |
+          pip install psutil cloudpickle ml_dtypes numpy packaging scipy tornado typing_extensions
       - name: Test
         shell: cmd /C call {0}
         run: >-
           python -m pytest -v tests/python/all-platform-minimal-test
-
-  # Disabled due to https://github.com/apache/tvm/issues/13950
-  # Windows-Static-Runtime:
-  #   if: ${{ github.repository == 'apache/tvm' }}
-  #   runs-on: windows-2019
-  #   steps:
-  #     - uses: actions/checkout@v2
-  #       with:
-  #         submodules: 'recursive'
-  #     - name: Set up environment
-  #       uses: ./.github/actions/setup
-  #     - name: Build static TVM runtime
-  #       shell: bash -l {0}
-  #       run: |
-  #         tests/scripts/task_config_build_static.sh build
-  #         cd build
-  #         cmake .. -A x64 -DCMAKE_CONFIGURATION_TYPES="Release"
-  #         cmake --build . --config Release --target runtime
-
-  Linux-Static-Runtime:
-    if: ${{ github.repository == 'apache/tvm' }}
-    runs-on: Ubuntu-20.04
-    steps:
-      - uses: actions/checkout@v2
-        with:
-          submodules: 'recursive'
-      - name: Set up environment
-        uses: ./.github/actions/setup
-      - name: Build static TVM runtime
-        shell: bash -l {0}
-        run: |
-          tests/scripts/task_config_build_static.sh build
-          cd build
-          cmake ..
-          cmake --build . --config Release --target runtime
-
-  Android:
-    if: ${{ github.repository == 'apache/tvm' }}
-    runs-on: Ubuntu-20.04
-    steps:
-      - uses: actions/checkout@v2
-        with:
-          submodules: 'recursive'
-      - name: Set up environment
-        uses: ./.github/actions/setup
-      - name: Set up java
-        uses: actions/setup-java@v3
-        with:
-          distribution: 'zulu'
-          java-version: '11'
-      - name: Build TVM
-        shell: bash -l {0}
-        run: |
-          mkdir build
-          cd build
-          ../tests/scripts/task_config_build_jvm.sh .
-          cmake ..
-          make
-      - name: Build TVM4J
-        run: |
-          make jvmpkg
-      - name: Build android_rpc
-        working-directory: apps/android_rpc
-        run: |
-          set -eux
-          export PATH="${ANDROID_NDK_LATEST_HOME}:$PATH"
-          gradle clean build
-      - name: Upload android_rpc APK
-        uses: actions/upload-artifact@v4
-        with:
-          name: android_rpc-debug.apk
-          path: ./apps/android_rpc/app/build/outputs/apk/debug/app-debug.apk

.github/workflows/nightly_docker_update.yml

@@ -1,8 +1,6 @@
 
 name: Nightly Docker Update
 on:
-  schedule:
-    - cron: "0 0 * * *"
   workflow_dispatch:
 
 concurrency:
@@ -20,9 +18,9 @@ jobs:
       pull-requests: write
       statuses: write
     if: github.repository == 'apache/tvm'
-    runs-on: ubuntu-20.04
+    runs-on: ubuntu-latest
     steps:
-      - uses: actions/checkout@v2
+      - uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd # v6.0.2
       - name: Open PR to update Docker images
         env:
           GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}