Test yb pipeline

.github/CODEOWNERS

@@ -46,3 +46,20 @@ lib/Dialect/TritonGPU/Transforms/TritonGPUConversion.cpp @ptillet
 # third_party
 # -----------
 third_party/amd/ @antiagainst @zhanglx13
+third_party/proton/ @Jokeren @crobeck @fywkevin
+
+# -----------
+# gluon
+# -----------
+python/triton/experimental/gluon/ @peterbell10
+python/src/gluon_ir.cc @peterbell10
+python/test/gluon @peterbell10
+test/Gluon @peterbell10
+include/triton/Dialect/Gluon @peterbell10
+lib/Dialect/Gluon @peterbell10
+
+# -----------
+# Linear Layouts
+# -----------
+lib/Tools/ @lezcano
+lib/Dialect/TritonGPU/IR/LinearLayoutConversions.cpp @lezcano

.github/ISSUE_TEMPLATE/bug.yml

@@ -0,0 +1,48 @@
+name: Report a bug
+description: Report triton failing to compile a kernel, or giving incorrect results
+labels: ["bug"]
+
+body:
+- type: markdown
+  attributes:
+    value: |
+      #### Disclaimer
+      The core triton team is small and has very limited capacity. We may not have time to look into your report.
+      For the best results, please:
+        - Avoid submitting duplicates. Search through [the existing and past issues](https://github.com/triton-lang/triton/issues?q=is%3Aissue+sort%3Acreated-desc+) first to see if it's been reported previously.
+        - Check if the issue persists with a build from the latest source.
+        - Provide all relevant information in the initial report, to prevent unnecessary back and forth discussion.
+        - If you can, try to diagnose and/or fix the issue yourself. We welcome high quality contributions.
+- type: textarea
+  attributes:
+    label: Describe the bug
+    description: |
+      Please provide a clear and concise description of what the bug is.
+
+      If relevant, add a [minimal complete example](https://stackoverflow.com/help/minimal-reproducible-example) that reproduces the bug. It is very important for the snippet to be as simple as possible, so please take time to trim down any irrelevant code to help us debug efficiently. We are going to copy-paste your code and we expect to get the same result as you did, so include both the kernel and launching code as well as any relevant imports.
+
+      If the code is too long (hopefully, it isn't), feel free to put it in a public gist and link it in the issue: https://gist.github.com.
+
+      Please also paste or describe the results you observe instead of the expected results. If you observe an error, please paste the error message including the **full** traceback of the exception. It may be relevant to wrap error messages in ```` ```triple quotes blocks``` ````.
+    placeholder: |
+      A clear and concise description of what the bug is.
+
+      ```python
+      # Sample code to reproduce the problem
+      ```
+
+      ```
+      The error message you got, with the full traceback.
+      ```
+  validations:
+    required: true
+- type: textarea
+  attributes:
+    label: Environment details
+    description: |
+      Please include any relevant context about how you're running the reproducer e.g. which version of triton, and what GPU you are using.
+    placeholder: |
+        Triton: ...
+        GPU: ...
+  validations:
+    required: true

.github/ISSUE_TEMPLATE/config.yml

@@ -0,0 +1,5 @@
+blank_issues_enabled: true
+contact_links:
+  - name: Community help
+    url: https://discord.gg/gpumode
+    about: GPU-mode discord community has a triton channel which is a great resource for help writing/learning triton

.github/ISSUE_TEMPLATE/performance.yml

@@ -0,0 +1,44 @@
+name: Report a performance issue
+description: Report cases where triton is generating sub-optimal (but functionally correct) PTX/LLVM IR
+labels: ["performance"]
+
+body:
+- type: markdown
+  attributes:
+    value: |
+      #### Disclaimer
+      The core triton team is small and has very limited capacity. We may not have time to look into your report.
+      For the best results, please:
+        - Avoid submitting duplicates. Search through [the existing and past issues](https://github.com/triton-lang/triton/issues?q=is%3Aissue+sort%3Acreated-desc+) first to see if it's been reported previously.
+        - Check if the issue persists with a build from the latest source.
+        - Provide all relevant information in the initial report, to prevent unnecessary back and forth discussion.
+        - If you can, try to diagnose and/or fix the issue yourself. We welcome high quality contributions.
+- type: textarea
+  attributes:
+    label: Describe the issue
+    description: |
+      Please provide a clear and concise description of the issue.
+
+      Include a [minimal complete example](https://stackoverflow.com/help/minimal-reproducible-example) that reproduces the issue. It is very important for the snippet to be as simple as possible, so please take time to trim down any irrelevant code to help us debug efficiently. We are going to copy-paste your code and we expect to get the same result as you did.
+
+      A reproducer could be a python program that runs a triton kernel and prints out the relevant suboptimal IR, or an IR file with an accompanying triton-opt command.
+
+      If the code is too long (hopefully, it isn't), feel free to put it in a public gist and link it in the issue: https://gist.github.com.
+    placeholder: |
+      A clear and concise description of the issue.
+
+      ```python
+      # Sample code to reproduce the problem
+      ```
+  validations:
+    required: true
+- type: textarea
+  attributes:
+    label: Environment details
+    description: |
+      Please include any relevant context about how you're running the reproducer e.g. which version of triton, and what GPU you are using.
+    placeholder: |
+        Triton: ...
+        GPU: ...
+  validations:
+    required: true

.github/PULL_REQUEST_TEMPLATE.md

@@ -1,11 +1,14 @@
+<!---
 The core Triton is a small number of people, and we receive many PRs (thank
 you!).  To help us review your code more quickly, **if you are a new
 contributor (less than 3 PRs merged) we ask that you complete the following
 tasks and include the filled-out checklist in your PR description.**
 
 Complete the following tasks before sending your PR, and replace `[ ]` with
 `[x]` to indicate you have done them.
+-->
 
+# New contributor declaration
 - [ ] I am not making a trivial change, such as fixing a typo in a comment.
 
 - [ ] I have written a PR description following these

.github/workflows/build-docker-image.yml

@@ -1,94 +1,80 @@
 name: Release Triton Ascend Images
 
 on:
-  pull_request:
-    types: [opened, reopened, synchronize]
-    branches: [main]
   push:
     branches: [main]
+    paths:
+      - 'docker/Dockerfile'
+      - 'Makefile'
+      - 'requirements.txt'
+      - 'requirements_dev.txt'
+      - '.github/workflows/build-docker-image.yml'
+  pull_request:
+    branches: [main]
+    paths:
+      - 'docker/Dockerfile'
+      - 'Makefile'
+      - 'requirements.txt'
+      - 'requirements_dev.txt'
+      - '.github/workflows/build-docker-image.yml'
 
 permissions:
   contents: read
+  actions: write
 
 jobs:
-  build-release-images:
-    name: Build and Push Docker Images (${{ matrix.npu_type }})
+  build-images:
+    name: Build ${{ matrix.cann_version }}-${{ matrix.chip_type }}-${{ matrix.os }}-py${{ matrix.python_version }}
     runs-on: ubuntu-22.04
 
     strategy:
       fail-fast: false
       matrix:
-        npu_type: ['910b', 'A3']
-    env:
-      QUAY_USER: ${{ secrets.QUAY_USER }}
-      QUAY_PASSWD: ${{ secrets.QUAY_PASSWD }}
-      HEAD_COMMIT: ${{ github.sha }}
+        cann_version: [8.5.0, 9.0.0-beta.2]
+        chip_type: [910b, a3]
+        os: [ubuntu22.04]
+        python_version: ["3.10", "3.11"]
+
     steps:
       - name: Checkout code
         uses: actions/checkout@v4
-        with:
-          fetch-depth: 0
 
       - name: Set up QEMU
         uses: docker/setup-qemu-action@v3
 
       - name: Set up Docker Buildx
         uses: docker/setup-buildx-action@v3
 
-      - name: Buildx info
-        run: |
-          echo "Checking Docker Buildx version and builders..."
-          docker buildx version
-          docker buildx ls
+      - name: Login to Quay.io
+        if: github.event_name == 'push'
+        uses: docker/login-action@v3
+        with:
+          registry: quay.io
+          username: ${{ secrets.QUAY_USER }}
+          password: ${{ secrets.QUAY_PASSWD }}
 
-      - name: Build and push release images
-        shell: bash
+      - name: Generate tags
+        id: tags
         run: |
-          if [ -z "${QUAY_USER}" ] || [ -z "${QUAY_PASSWD}" ]; then
-            echo "Please set QUAY_USER and QUAY_PASSWD secrets before building images."
-            exit 1
-          fi
-
-          if [ -n "${HEAD_COMMIT}" ]; then
-            GIT_COMMIT_SHORT="$(git rev-parse --short "${HEAD_COMMIT}")"
-          else
-            GIT_COMMIT_SHORT="$(git rev-parse --short HEAD)"
-          fi
-
-          echo "${QUAY_PASSWD}" | docker login -u "${QUAY_USER}" --password-stdin quay.io
-
-          BUILD_IMAGE=0
-          if [ -z "${BASE_COMMIT}" ]; then
-            echo "BASE_COMMIT not set. Forcing Docker image build..."
-            BUILD_IMAGE=1
-          elif [ -n "${filenames}" ] && echo "${filenames}" | grep -Eq '\b(docker/Dockerfile_build|Makefile|requirements(_dev)?\.txt)\b'; then
-            echo "Relevant files changed. Building Docker image..."
-            BUILD_IMAGE=1
-          else
-            echo "No relevant changes since BASE_COMMIT. Skipping Docker image build."
-          fi
+          CANN_TAG="${{ matrix.cann_version }}-${{ matrix.chip_type }}-${{ matrix.os }}-py${{ matrix.python_version }}"
+          GIT_COMMIT_SHORT="$(git rev-parse --short HEAD)"
+          TAG_COMMIT="${CANN_TAG}-${GIT_COMMIT_SHORT}"
+          TAG_LATEST="${CANN_TAG}-latest"
+          echo "cann_tag=${CANN_TAG}" >> $GITHUB_OUTPUT
+          echo "tag_commit=${TAG_COMMIT}" >> $GITHUB_OUTPUT
+          echo "tag_latest=${TAG_LATEST}" >> $GITHUB_OUTPUT
 
-          if [ "$BUILD_IMAGE" -eq 1 ]; then
-            docker buildx build --platform "linux/amd64,linux/arm64" \
-              --build-arg NPU_TYPE=${{ matrix.npu_type }} \
-              -f docker/Dockerfile_build --push \
-              -t "quay.io/ascend/triton:dev-build-${GIT_COMMIT_SHORT}-${{ matrix.npu_type }}" .
-          fi
-
-          BUILD_IMAGE=0
-          if [ -z "${BASE_COMMIT}" ]; then
-            echo "BASE_COMMIT not set. Forcing Docker image build..."
-            BUILD_IMAGE=1
-          elif [ -n "${filenames}" ] && echo "${filenames}" | grep -Eq '\b(docker/Dockerfile|Makefile|requirements(_dev)?\.txt)\b'; then
-            echo "Relevant files changed. Building Docker image..."
-            BUILD_IMAGE=1
-          else
-            echo "No relevant changes since BASE_COMMIT. Skipping Docker image build."
-          fi
-
-          if [ "$BUILD_IMAGE" -eq 1 ]; then
-            docker buildx build --platform "linux/amd64,linux/arm64" \
-              --build-arg NPU_TYPE=${{ matrix.npu_type }} \
-              -f docker/Dockerfile --push \
-              -t "quay.io/ascend/triton:dev-${GIT_COMMIT_SHORT}-${{ matrix.npu_type }}" .
-          fi
+      - name: Build and push
+        uses: docker/build-push-action@v6
+        with:
+          context: .
+          file: docker/Dockerfile
+          platforms: linux/amd64,linux/arm64
+          push: ${{ github.event_name == 'push' }}
+          tags: |
+            quay.io/ascend/triton:${{ steps.tags.outputs.tag_commit }}
+            quay.io/ascend/triton:${{ steps.tags.outputs.tag_latest }}
+          build-args: |
+            CANN_BASE_IMAGE=quay.io/ascend/cann:${{ steps.tags.outputs.cann_tag }}
+          cache-from: type=gha
+          cache-to: type=gha,mode=max

.github/workflows/ci.yml

@@ -2,7 +2,26 @@ name: Integration Tests
 on:
   workflow_dispatch:
   pull_request:
-    branches-ignore: ['llvm-**']
+    paths-ignore:
+      - 'docs/**'
+      - '**/*.md'
+      - 'LICENSE'
+      - 'MANIFEST.in'
+      - '.gitignore'
+      - '.github/CODEOWNERS'
+      - '.github/dependabot.yml'
+      - 'cmake/llvm-hash.txt'
+      - 'third_party/ascend/llvm_patch/**'
+      - 'docker/**'
+      - 'requirements.txt'
+      - 'requirements_dev.txt'
+      - '.github/workflows/build-docker-image.yml'
+      - '.github/workflows/create_release.yml'
+      - '.github/workflows/documentation.yml'
+      - '.github/workflows/llvm-build.yml'
+      - '.github/workflows/llvm-build/**'
+      - '.github/workflows/pre-commit.yml'
+      - '.github/workflows/runner-preparation.yml'
   merge_group:
     branches: [main, 'dev-**']
     types: [checks_requested]
@@ -18,12 +37,24 @@ jobs:
   runner-preparation:
     uses: ./.github/workflows/runner-preparation.yml
 
-  pre-commit:
-    uses: ./.github/workflows/pre-commit.yml
+  # pre-commit:
+  #   uses: ./.github/workflows/pre-commit.yml
 
-  integration-tests-ascend:
+  build-wheels:
     needs: runner-preparation
     if: needs.runner-preparation.outputs.matrix-ASCEND != ''
+    uses: ./.github/workflows/wheels.yml
+    with:
+      ci_only: true
+    secrets: inherit
+    permissions:
+      id-token: write
+      contents: read
+
+  integration-tests-ascend:
+    needs: [runner-preparation, build-wheels]
+    if: needs.runner-preparation.outputs.matrix-ASCEND != ''
     uses: ./.github/workflows/integration-tests-ascend.yml
     with:
       matrix: ${{ needs.runner-preparation.outputs.matrix-ASCEND }}
+      use_prebuilt_wheel: true

.github/workflows/create_release.yml

@@ -26,7 +26,7 @@ jobs:
     outputs:
       release_name: "${{ steps.release_name.outputs.name }}"
     steps:
-      - uses: actions/checkout@v6
+      - uses: actions/checkout@v4
         with:
           show-progress: false
           submodules: 'recursive'
@@ -47,32 +47,28 @@ jobs:
             # strip trailing v from tag name
             tag_or_branch="${tag_or_branch#v}"
             # important: version must be fixed in setup.py
-            sed -i -e "s:^TRITON_VERSION = .*:TRITON_VERSION = '${tag_or_branch}':" python/setup.py || exit 1
+            sed -i -e "s:^TRITON_VERSION = .*:TRITON_VERSION = '${tag_or_branch}':" setup.py || exit 1
           fi
-          echo "RELEASE_NAME=triton-ascend$tag_or_branch" >> "$GITHUB_ENV"
-
+          echo "RELEASE_NAME=triton-ascend-$tag_or_branch" >> "$GITHUB_ENV"
       - name: Create source distribution
-        working-directory: python
         run: |
-          python -m pip install --upgrade pip -qq || exit 1
-          python -m pip install "setuptools<70" wheel build -qq || exit 1
+          python -m pip install --upgrade pip || exit 1
+          python -m pip install "setuptools<70" wheel build || exit 1
           python -m build -s || exit 1
           cd dist || exit 1
           release_file=( *.tar.gz )
           echo "RELEASE_FILE=${release_file}" >> "$GITHUB_ENV"
-
       - name: Upload source distribution for release
         if: ${{ github.event_name == 'release' }}
         uses: softprops/action-gh-release@v2
         with:
-          files: python/dist/${{ env.RELEASE_FILE }}
-
+          files: dist/${{ env.RELEASE_FILE }}
       - name: Upload source distribution to GHA artifacts for release tags
         if: ${{ github.event_name == 'push' && startsWith(github.ref, 'refs/tags/v') && contains(github.ref, 'rc') }}
         uses: actions/upload-artifact@v4.4.0
         with:
           name: ${{ env.RELEASE_FILE }}
-          path: python/dist/${{ env.RELEASE_FILE }}
+          path: dist/${{ env.RELEASE_FILE }}
       - name: Set output
         id: release_name
         run: echo "name=release_name::${{ env.RELEASE_NAME }}.tar.gz" >> "${GITHUB_OUTPUT}"