Poc tle gpu tileir

.github/CODEOWNERS

@@ -2,47 +2,14 @@
 # the repo. Unless a later match takes precedence,
 # @global-owner1 and @global-owner2 will be requested for
 # review when someone opens a pull request.
-*       @ptillet
+*       @zhzhcookie
 
 # --------
-# Analyses
+# Setup
 # --------
-# Alias analysis
-include/triton/Analysis/Alias.h @Jokeren
-lib/Analysis/Alias.cpp @Jokeren
-# Allocation analysis
-include/triton/Analysis/Allocation.h @Jokeren
-lib/Analysis/Allocation.cpp @Jokeren
-# Membar analysis
-include/triton/Analysis/Membar.h @Jokeren
-lib/Analysis/Membar.cpp @Jokeren
-# AxisInfo analysis
-include/triton/Analysis/AxisInfo.h @ptillet
-lib/Analysis/AxisInfo.cpp @ptillet
-# Utilities
-include/triton/Analysis/Utility.h @Jokeren
-lib/Analysis/Utility.cpp @Jokeren
-
-# ----------
-# Dialects
-# ----------
-# Pipeline pass
-lib/Dialect/TritonGPU/Transforms/Pipeline.cpp @ptillet
-# Prefetch pass
-lib/Dialect/TritonGPU/Transforms/Prefetch.cpp @ptillet
-# Coalesce pass
-lib/Dialect/TritonGPU/Transforms/Coalesce.cpp @ptillet
-# Layout simplification pass
-lib/Dialect/TritonGPU/Transforms/Combine.cpp @ptillet
-
-# -----------
-# Conversions
-# -----------
-# TritonToTritonGPU
-include/triton/Conversion/TritonToTritonGPU/ @ptillet
-lib/Dialect/TritonGPU/Transforms/TritonGPUConversion.cpp @ptillet
+python/setup.py @Galaxy1458
+python/setup_tools/** @Galaxy1458
 
 # -----------
 # third_party
 # -----------
-third_party/amd/ @antiagainst @zhanglx13

.github/PULL_REQUEST_TEMPLATE.md

@@ -1,27 +1,16 @@
-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.
-
-- [ ] I am not making a trivial change, such as fixing a typo in a comment.
-
-- [ ] I have written a PR description following these
-  [rules](https://cbea.ms/git-commit/#why-not-how).
-
-- [ ] I have run `pre-commit run --from-ref origin/main --to-ref HEAD`.
-
-- Select one of the following.
-  - [ ] I have added tests.
-    - `/test` for `lit` tests
-    - `/unittest` for C++ tests
-    - `/python/test` for end-to-end tests
-  - [ ] This PR does not need a test because `FILL THIS IN`.
-
-- Select one of the following.
-  - [ ] I have not added any `lit` tests.
-  - [ ] The `lit` tests I have added follow these [best practices](https://mlir.llvm.org/getting_started/TestingGuide/#filecheck-best-practices),
-    including the "tests should be minimal" section. (Usually running Python code
-    and using the instructions it generates is not minimal.)
+<!-- PR Title
+[component] brief description
+  component options:
+    - BUILD
+    - CI/CD
+    - DOC
+    - FRONTEND
+    - BACKEND
+    - AUTOTUNER
+    - CACHE
+    - LAYOUTS
+    - PIPELINE
+    - PROTON
+    - TEST
+    - OTHER
+-->

.github/workflows/ascend-build-and-test.yml

@@ -0,0 +1,105 @@
+name: Ascend3.2-Build-And-Test
+
+on:
+  push:
+    branches: [ "triton_v3.2.x" ]
+  pull_request:
+    branches: [ "triton_v3.2.x" ]
+
+concurrency:
+  group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref }}
+  cancel-in-progress: true
+
+jobs:
+  ascend-build-and-test:
+    runs-on: ascend3.2
+    if: ${{ github.repository == 'FlagTree/flagtree' || github.repository == 'flagos-ai/flagtree' }}
+    steps:
+      - name: Setup environment
+        shell: bash
+        run: |
+          source ~/env.sh
+          env | grep -E '^(http_proxy|https_proxy|all_proxy|no_proxy)=' >> $GITHUB_ENV || true
+
+      - name: Smart Checkout
+        uses: flagos-ai/FlagTree/.github/actions/smart-checkout@main
+        with:
+          checkout_version: 'v6'
+
+      - name: Check if backend-relevant files changed
+        id: check_backend
+        uses: flagos-ai/FlagTree/.github/actions/check-backend-changed@main
+        with:
+          backend: ascend
+
+      - name: FlagTree Editable Build And Test on Ascend
+        if: steps.check_backend.outputs.should_skip != 'true'
+        shell: bash
+        run: |
+          set -x
+          export FLAGTREE_BACKEND=ascend
+          source ~/env.sh
+          python3 -m pip uninstall flagtree -y
+          pushd python
+          MAX_JOBS=32 python3 -m pip install -e . --no-build-isolation
+          popd
+          source /usr/local/Ascend/ascend-toolkit/set_env.sh
+          python3 -c "import triton"
+          python3 -m pip uninstall flagtree -y
+
+      - name: FlagTree Build on Ascend
+        if: steps.check_backend.outputs.should_skip != 'true'
+        shell: bash
+        run: |
+          set -x
+          export FLAGTREE_BACKEND=ascend
+          source ~/env.sh
+          cd python
+          MAX_JOBS=32 python3 -m pip install . --no-build-isolation
+
+      - name: FlagTree Test on Ascend
+        if: steps.check_backend.outputs.should_skip != 'true'
+        shell: bash
+        run: |
+          set -x
+          source /usr/local/Ascend/ascend-toolkit/set_env.sh
+          # tutorials
+          pushd third_party/ascend/tutorials
+          python3 01-vector-add.py
+          python3 02-fused-softmax.py
+          python3 03-layer-norm.py
+          python3 04-fused-attention.py
+          python3 05-matrix-multiplication.py
+          python3 06-demo-autotune.py
+          python3 07-profiler.py
+          python3 08-demo-libentry.py
+          python3 09-gather.py >output-09-gather.log
+          python3 10-gather_sorted.py >output-10-gather_sorted.log
+          python3 11-rab_time.py
+          #python3 12-hstu_attention.py
+          python3 13-matrix-multiplication-optimized.py
+          python3 14-accuracy-comparison.py
+          #python3 15-embedding_gather_demo.py
+          popd
+          # pytest_ut
+          pushd third_party/ascend/unittest/pytest_ut
+          python3 -m pytest . \
+              --ignore=test_add_mindspore.py \
+              --ignore=test_softmax_mindspore.py \
+              --ignore=test_assume.py \
+              --ignore=test_index_select.py
+          popd
+          # flagtree tle test
+          pushd python/test/tle
+          python3 test_vec_add.py
+          python3 test_vec_add_2d.py
+          python3 test_vec_add_mix.py
+          python3 test_vec_mathOps.py
+          python3 test_tle_with_hints.py
+          popd
+          # flagtree tle tutorial
+          pushd python/tutorials/tle
+          python3 06-swiglu.py
+          python3 07-kv-rmsnorm-rope-cache.py
+          pytest -s -k "accuracy" 08-add-rms-norm-bias.py
+          popd

.github/workflows/ascend3.2-delivery.yml

@@ -0,0 +1,44 @@
+name: CD-Ascend3.2
+
+on:
+  workflow_dispatch:
+    inputs:
+      WHL_VER:
+        description: '0.6.0.dev1 < 0.6.0a1 < 0.6.0b1 < 0.6.0rc1 < 0.6.0 < 0.6.0.post1'
+        required: true
+        default: '0.6.0'
+      PYTHON_VER:
+        required: true
+        default: '3.11'
+      PLATFORM:
+        required: true
+        default: 'aarch64'
+      FLAGTREE_PYPI_KEY:
+        required: false
+  workflow_call:
+    inputs:
+      WHL_VER:
+        type: string
+        required: true
+        default: '0.6.0'
+      PYTHON_VER:
+        type: string
+        required: true
+        default: '3.11'
+      PLATFORM:
+        type: string
+        required: true
+        default: 'aarch64'
+      FLAGTREE_PYPI_KEY:
+        type: string
+        required: false
+
+jobs:
+  call-delivery:
+    uses: flagos-ai/flagtree/.github/workflows/ascend3.2-delivery.yml@main
+    with:
+      WHL_VER: ${{ inputs.WHL_VER }}
+      PYTHON_VER: ${{ inputs.PYTHON_VER }}
+      PLATFORM: ${{ inputs.PLATFORM }}
+      FLAGTREE_PYPI_KEY: ${{ inputs.FLAGTREE_PYPI_KEY }}
+    secrets: inherit

.github/workflows/cambricon-build-and-test.yml

@@ -0,0 +1,36 @@
+name: Cambricon-Build-And-Test
+
+on:
+  push:
+    branches: [ "main" ]
+  pull_request:
+    branches: [ "main" ]
+
+concurrency:
+  group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref }}
+  cancel-in-progress: true
+
+jobs:
+  cambricon-build-and-test:
+    runs-on: cambricon
+    if: ${{ github.repository == 'FlagTree/flagtree' || github.repository == 'flagos-ai/flagtree' }}
+    steps:
+      - name: Setup environment
+        shell: bash
+        run: |
+          source ~/env.sh
+          env | grep -E '^(http_proxy|https_proxy|all_proxy|no_proxy)=' >> $GITHUB_ENV || true
+
+      - name: Smart Checkout
+        uses: flagos-ai/FlagTree/.github/actions/smart-checkout@main
+        with:
+          checkout_version: 'v6'
+
+      - name: FlagTree Build on Cambricon
+        shell: bash
+        run: |
+          set -x
+          pip uninstall -y triton
+          export FLAGTREE_BACKEND=cambricon
+          cd python
+          MAX_JOBS=8 python3 -m pip install . --no-build-isolation

.github/workflows/code-format-check.yml

@@ -0,0 +1,25 @@
+name: Code-Format-Check
+
+on:
+  schedule:
+    - cron: '0 21 * * *'
+  push:
+    branches: [ "main", "triton_v3.2.x", "triton_v3.3.x", "triton_v3.4.x", "triton_v3.5.x"]
+  pull_request:
+    branches: [ "main", "triton_v3.2.x", "triton_v3.3.x", "triton_v3.4.x", "triton_v3.5.x"]
+
+concurrency:
+  group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref }}
+  cancel-in-progress: true
+
+jobs:
+  pre-commit:
+    runs-on: ubuntu-latest
+    steps:
+    - name: Checkout code
+      id: checkout
+      uses: actions/checkout@v6
+      with:
+        token: ${{ secrets.BOT_PAT_TOKEN || github.token }}
+    - name: Code format check
+      uses: flagos-ai/FlagTree/.github/actions/code-format-check@main

.github/workflows/code_scan.yml

@@ -0,0 +1,57 @@
+name: code-scan
+
+on:
+  push:
+    branches: [ "main" ]
+  pull_request:
+    branches: [ "main" ]
+
+jobs:
+  scan-code-and-report:
+    runs-on: scan
+    if: ${{ github.repository == 'FlagTree/flagtree' }}
+    concurrency:
+      group: scan-code-and-report-${{ github.event.pull_request.number || github.ref }}
+      cancel-in-progress: true
+
+    steps:
+      - name: Setup environment
+        shell: bash
+        run: |
+          source ~/env.sh
+          env | grep -E '^(http_proxy|https_proxy|all_proxy|no_proxy)=' >> $GITHUB_ENV || true
+
+      - name: Smart Checkout
+        uses: flagos-ai/FlagTree/.github/actions/smart-checkout@main
+        with:
+          checkout_version: 'v6'
+
+      - name: Scan flagtree repo
+        shell: bash
+        run: |
+          set -x
+          git config --global --add safe.directory ../flagtree
+          if [ "${{ github.event_name }}" == "pull_request" ]; then
+            echo "This is a pull request event. PR number is ${{ github.event.pull_request.number }}"
+            PR_ID=${{ github.event.pull_request.number }}
+          elif [ "${{ github.event_name }}" == "push" ]; then
+            PR_NUMBER=$(git log -1 --pretty=format:'%s' | grep -oE '#[0-9]+' | grep -oE '[0-9]+')
+            echo "This is a push event. The relate PR number is ${PR_NUMBER}"
+            PR_ID=${PR_NUMBER}
+          fi
+
+          python /work/flag_tree/code_can/scan_code.py --pr ${PR_ID} --hash ${GITHUB_SHA} --attempt ${GITHUB_RUN_ATTEMPT}
+
+      - name: Code Scan Report
+        shell: bash
+        run: |
+          set -x
+          git config --global --add safe.directory ../flagtree
+          if [ "${{ github.event_name }}" == "pull_request" ]; then
+            PR_ID=${{ github.event.pull_request.number }}
+          elif [ "${{ github.event_name }}" == "push" ]; then
+            PR_NUMBER=$(git log -1 --pretty=format:'%s' | grep -oE '#[0-9]+' | grep -oE '[0-9]+')
+            PR_ID=${PR_NUMBER}
+          fi
+          echo "Repo Scan report:"
+          echo "http://120.92.44.177/tree_scan_report/PR${PR_ID}-${GITHUB_SHA}-${GITHUB_RUN_ATTEMPT}.html"