Upgrade amd-integration branch to v0.5.3 tag from upstream.

.devcontainer/cuda/devcontainer.json → .devcontainer/cu126/devcontainer.json

@@ -1,8 +1,8 @@
 {
     "name": "CUDA Development Container",
     "build": {
-        "dockerfile": "Dockerfile",
-        "context": "."
+        "dockerfile": "../../docker/Dockerfile.cu126.dev",
+        "context": "../../"
     },
     "runArgs": [
         "--gpus=all"
@@ -12,8 +12,10 @@
             "extensions": [
                 "llvm-vs-code-extensions.vscode-clangd",
                 "ms-python.python",
-                "ms-python.black-formatter",
-                "nvidia.nsight-vscode-edition"
+                "nvidia.nsight-vscode-edition",
+                "xaver.clang-format",
+                "charliermarsh.ruff",
+                "matangover.mypy"
             ]
         }
     },

.devcontainer/cu128/devcontainer.json

@@ -0,0 +1,26 @@
+{
+    "name": "CUDA Development Container",
+    "build": {
+        "dockerfile": "../../docker/Dockerfile.cu128.dev",
+        "context": "../../"
+    },
+    "runArgs": [
+        "--gpus=all"
+    ],
+    "customizations": {
+        "vscode": {
+            "extensions": [
+                "llvm-vs-code-extensions.vscode-clangd",
+                "ms-python.python",
+                "nvidia.nsight-vscode-edition",
+                "xaver.clang-format",
+                "charliermarsh.ruff",
+                "matangover.mypy"
+            ]
+        }
+    },
+    "mounts": [
+        "type=bind,source=${localEnv:HOME}/.ssh,target=/home/devuser/.ssh,readonly"
+    ],
+    "remoteUser": "devuser"
+}

.devcontainer/cu129/devcontainer.json

@@ -0,0 +1,26 @@
+{
+    "name": "CUDA Development Container",
+    "build": {
+        "dockerfile": "../../docker/Dockerfile.cu129.dev",
+        "context": "../../"
+    },
+    "runArgs": [
+        "--gpus=all"
+    ],
+    "customizations": {
+        "vscode": {
+            "extensions": [
+                "llvm-vs-code-extensions.vscode-clangd",
+                "ms-python.python",
+                "nvidia.nsight-vscode-edition",
+                "xaver.clang-format",
+                "charliermarsh.ruff",
+                "matangover.mypy"
+            ]
+        }
+    },
+    "mounts": [
+        "type=bind,source=${localEnv:HOME}/.ssh,target=/home/devuser/.ssh,readonly"
+    ],
+    "remoteUser": "devuser"
+}

.devcontainer/cu130/devcontainer.json

@@ -0,0 +1,26 @@
+{
+    "name": "CUDA Development Container",
+    "build": {
+        "dockerfile": "../../docker/Dockerfile.cu130.dev",
+        "context": "../../"
+    },
+    "runArgs": [
+        "--gpus=all"
+    ],
+    "customizations": {
+        "vscode": {
+            "extensions": [
+                "llvm-vs-code-extensions.vscode-clangd",
+                "ms-python.python",
+                "nvidia.nsight-vscode-edition",
+                "xaver.clang-format",
+                "charliermarsh.ruff",
+                "matangover.mypy"
+            ]
+        }
+    },
+    "mounts": [
+        "type=bind,source=${localEnv:HOME}/.ssh,target=/home/devuser/.ssh,readonly"
+    ],
+    "remoteUser": "devuser"
+}

.github/CODEOWNERS

@@ -0,0 +1,45 @@
+# Code Owners File
+# Generated automatically from git history analysis
+# Analysis period: 180 days
+# Minimum commits threshold: 1
+
+benchmarks/ @bkryu @jiahanc @cyx-6 @yzh119 @nv-yunzheq
+benchmarks/routines/ @bkryu @nv-yunzheq @jiahanc @cyx-6 @nvmbreughe
+ci/ @cyx-6 @yzh119 @nvmbreughe
+ci/scripts/ @cyx-6
+ci/scripts/jenkins/ @cyx-6
+csrc/ @wenscarl @yzh119 @cyx-6 @djmmoss @nv-yunzheq
+csrc/fused_moe/ @nv-yunzheq @yzh119 @yongwww @djmmoss @cyx-6
+csrc/fused_moe/cutlass_backend/ @nv-yunzheq @yzh119 @yongwww @djmmoss @cyx-6
+csrc/nv_internal/ @wenscarl @djmmoss @nv-yunzheq @yongwww @cyx-6
+csrc/nv_internal/cpp/ @wenscarl @bkryu @yongwww @djmmoss @joker-eph
+csrc/nv_internal/include/ @wenscarl @nv-yunzheq
+csrc/nv_internal/tensorrt_llm/ @wenscarl @djmmoss @nv-yunzheq @yongwww @cyx-6
+csrc/xqa/ @cyx-6 @yzh119
+docs/ @yzh119 @cyx-6 @wenscarl @nv-yunzheq @aleozlx
+flashinfer/ @yzh119 @cyx-6 @nvmbreughe @aleozlx @wenscarl
+flashinfer-cubin/ @yzh119 @cyx-6
+flashinfer-cubin/flashinfer_cubin/ @yzh119
+flashinfer-jit-cache/ @yzh119 @cyx-6
+flashinfer-jit-cache/flashinfer_jit_cache/ @yzh119
+flashinfer/comm/ @yzh119 @cyx-6 @nvmbreughe @wenscarl @djmmoss
+flashinfer/cudnn/ @Anerudhan @yzh119 @cyx-6 @Anerudhan
+flashinfer/cute_dsl/ @yzh119 @kaixih @Amir-19 @aleozlx
+flashinfer/dsv3_ops/ @nvmbreughe
+flashinfer/fused_moe/ @djmmoss @jiahanc @yzh119 @cyx-6 @aleozlx
+flashinfer/gemm/ @nvmbreughe
+flashinfer/jit/ @yzh119 @cyx-6 @aleozlx @jiahanc @nvmbreughe
+flashinfer/jit/attention/ @yzh119 @cyx-6 @Anerudhan @joker-eph
+flashinfer/jit/gemm/ @yzh119 @nv-yunzheq @jiahanc
+flashinfer/logits_processor/ @cyx-6 @yzh119
+flashinfer/profiler/ @cyx-6
+flashinfer/triton/ @nvmbreughe @cyx-6
+flashinfer/tuning_configs/ @kaixih
+include/ @yzh119 @jiahanc @nvmbreughe @IwakuraRein @bkryu
+include/flashinfer/ @yzh119 @jiahanc @nvmbreughe @IwakuraRein @bkryu
+include/flashinfer/attention/ @yzh119 @kahyunnam @joker-eph
+include/flashinfer/comm/ @yongwww @nvmbreughe @djmmoss @yzh119 @cyx-6
+include/flashinfer/gemm/ @ttyio @yongwww @yzh119 @nvmbreughe @aleozlx
+include/flashinfer/trtllm/ @jiahanc @joker-eph @aleozlx @yzh119 @wenscarl
+profiler/ @cyx-6
+scripts/ @yzh119 @nvmbreughe @dierksen @yongwww @bkryu

.github/workflows/build-doc.yml

@@ -5,6 +5,10 @@ on:
     branches:
       - main
 
+  pull_request:
+    branches:
+      - main
+
 # Sets permissions of the GITHUB_TOKEN to allow deployment to GitHub Pages
 permissions:
   contents: read
@@ -42,17 +46,18 @@ jobs:
         python -m pip install -r docs/requirements.txt
 
     - name: Bulid Documentation
-      if: github.ref == 'refs/heads/main'
       run: |
         cd docs
         ./build_docs.sh
 
     - name: Upload artifact
+      if: github.event_name == 'push' && github.ref == 'refs/heads/main'
       uses: actions/upload-pages-artifact@v3
       with:
         # Upload entire repository
         path: 'docs/_build/html'
 
     - name: Deploy to GitHub Pages
+      if: github.event_name == 'push' && github.ref == 'refs/heads/main'
       id: deployment
       uses: actions/deploy-pages@v4

.github/workflows/new-issue.yml

@@ -0,0 +1,20 @@
+name: Triage new issue
+
+on:
+    issues:
+      types:
+        - opened
+
+jobs:
+  label_issues:
+    runs-on: ubuntu-latest
+    permissions:
+      issues: write
+    steps:
+      - name: Add needs-triage label
+        run: gh issue edit "$NUMBER" --add-label "$LABELS"
+        env:
+          GH_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+          GH_REPO: ${{ github.repository }}
+          NUMBER: ${{ github.event.issue.number }}
+          LABELS: needs-triage