[BUILD] update flagtree backend to version3.4

.github/ISSUE_TEMPLATE/bug.yml

@@ -1,15 +1,15 @@
 name: Report a bug
-description: Report triton failing to compile a kernel, or giving incorrect results
+description: Report flagtree 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.
+      The core flagtree 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.
+        - Avoid submitting duplicates. Search 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.

.github/ISSUE_TEMPLATE/performance.yml

@@ -7,9 +7,9 @@ body:
   attributes:
     value: |
       #### Disclaimer
-      The core triton team is small and has very limited capacity. We may not have time to look into your report.
+      The core flagtree 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.
+        - Avoid submitting duplicates. Search 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.

.github/PULL_REQUEST_TEMPLATE.md

@@ -1,30 +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.
+<!-- PR Title
+[component] brief description
+  component options:
+    - BUILD
+    - CI/CD
+    - DOC
+    - FRONTEND
+    - BACKEND
+    - AUTOTUNER
+    - CACHE
+    - LAYOUTS
+    - PIPELINE
+    - PROTON
+    - TEST
+    - OTHER
 -->
-
-# 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
-  [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.)

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

@@ -0,0 +1,91 @@
+name: AIPU-Build-And-Test
+
+on:
+  push:
+    branches: [ "triton_v3.3.x" ]
+  pull_request:
+    branches: [ "triton_v3.3.x" ]
+
+concurrency:
+  group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref }}
+  cancel-in-progress: true
+
+jobs:
+  aipu-build-and-test:
+    runs-on: aipu
+    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: Checkout code (attempt 1)
+        id: checkout1
+        uses: actions/checkout@v5
+        continue-on-error: true
+
+      - name: Sleep before checkout2
+        if: steps.checkout1.outcome == 'failure'
+        run: |
+          echo "First checkout attempt failed. Sleeping for 120 seconds before retry..."
+          sleep 120
+
+      - name: Checkout code (attempt 2)
+        id: checkout2
+        if: steps.checkout1.outcome == 'failure'
+        uses: actions/checkout@v5
+        continue-on-error: true
+
+      - name: Sleep before final checkout
+        if: steps.checkout1.outcome == 'failure' && steps.checkout2.outcome == 'failure'
+        run: |
+          echo "Second checkout attempt failed. Sleeping for 180 seconds before final retry..."
+          sleep 180
+
+      - name: Checkout code (final attempt)
+        if: steps.checkout1.outcome == 'failure' && steps.checkout2.outcome == 'failure'
+        uses: actions/checkout@v5
+
+      - name: Verify checkout success
+        if: success()
+        run: echo "Checkout completed successfully"
+
+      - name: FlagTree Build on AIPU
+        shell: bash
+        run: |
+          set -x
+          pip uninstall -y triton
+          source ~/env_setup.sh
+          export FLAGTREE_BACKEND=aipu
+          cd python
+          MAX_JOBS=32 python3.10 -m pip install . --no-build-isolation -v
+
+      - name: FlagTree Test on AIPU
+        shell: bash
+        run: |
+          set -x
+          source ~/env_setup.sh
+          python3.10 third_party/aipu/python/test/test_01_vector_add.py
+          python3.10 third_party/aipu/python/test/test_01_vector_add_shared_memory.py
+          python3.10 third_party/aipu/python/test/test_01_vector_add_dma.py
+          python3.10 third_party/aipu/python/test/test_02_fused_softmax.py
+          python3.10 third_party/aipu/python/test/test_02_fused_softmax_shared_memory.py
+          python3.10 third_party/aipu/python/test/test_02_fused_softmax_dma.py
+          python3.10 third_party/aipu/python/test/test_libdevice_fmod.py
+          python3.10 third_party/aipu/python/test/test_libdevice_fmod_shared_memory.py
+          python3.10 third_party/aipu/python/test/test_libdevice_fmod_dma.py
+          python3.10 third_party/aipu/python/test/test_libdevice_pow.py
+          python3.10 third_party/aipu/python/test/test_libdevice_pow_shared_memory.py
+          python3.10 third_party/aipu/python/test/test_libdevice_pow_dma.py
+          python3.10 third_party/aipu/python/test/test_codegen_dma.py
+          python3.10 third_party/aipu/python/test/test_01_vector_add_structured_1d.py
+          python3.10 third_party/aipu/python/test/test_01_vector_add_structured_1d_shared_memory.py
+          python3.10 third_party/aipu/python/test/test_01_vector_add_structured_1d_dma.py
+          python3.10 third_party/aipu/python/test/min.py
+          python3.10 third_party/aipu/python/test/min_shared_memory.py
+          #python3.10 third_party/aipu/python/test/min_dma.py
+          python3.10 third_party/aipu/python/test/var_mean.py
+          python3.10 third_party/aipu/python/test/var_mean_shared_memory.py
+          #python3.10 third_party/aipu/python/test/var_mean_dma.py

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

@@ -0,0 +1,78 @@
+name: Ascend-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: ascend
+    steps:
+      - name: Checkout code (attempt 1)
+        id: checkout1
+        uses: actions/checkout@v5
+        continue-on-error: true
+
+      - name: Sleep before checkout2
+        if: steps.checkout1.outcome == 'failure'
+        run: |
+          echo "First checkout attempt failed. Sleeping for 120 seconds before retry..."
+          sleep 120
+
+      - name: Checkout code (attempt 2)
+        id: checkout2
+        if: steps.checkout1.outcome == 'failure'
+        uses: actions/checkout@v5
+        continue-on-error: true
+
+      - name: Sleep before final checkout
+        if: steps.checkout1.outcome == 'failure' && steps.checkout2.outcome == 'failure'
+        run: |
+          echo "Second checkout attempt failed. Sleeping for 180 seconds before final retry..."
+          sleep 180
+
+      - name: Checkout code (final attempt)
+        if: steps.checkout1.outcome == 'failure' && steps.checkout2.outcome == 'failure'
+        uses: actions/checkout@v5
+
+      - name: Verify checkout success
+        if: success()
+        run: echo "Checkout completed successfully"
+
+      - name: FlagTree Build on Ascend
+        shell: bash
+        run: |
+          set -x
+          pip uninstall -y triton
+          export FLAGTREE_BACKEND=ascend
+          source ~/env.sh
+          cd python
+          MAX_JOBS=32 python3 -m pip install . --no-build-isolation
+
+      - name: FlagTree Test on Ascend
+        shell: bash
+        run: |
+          set -x
+          source /usr/local/Ascend/ascend-toolkit/set_env.sh
+          python3 third_party/tests/ascend/vector-add.py
+          python3 third_party/ascend/examples/tutorials/01-vector-add.py
+          python3 third_party/ascend/examples/tutorials/02-fused-softmax.py
+          python3 third_party/ascend/examples/tutorials/03-layer-norm.py
+          #python3 third_party/ascend/examples/tutorials/04-fused-attention.py
+          python3 third_party/ascend/examples/tutorials/05-matrix-multiplication.py
+          python3 third_party/ascend/examples/tutorials/06-demo-autotune.py
+          python3 third_party/ascend/examples/tutorials/07-profiler.py
+          python3 third_party/ascend/examples/tutorials/08-demo-libentry.py
+          python3 third_party/ascend/examples/tutorials/09-gather.py
+          python3 third_party/ascend/examples/tutorials/10-gather_sorted.py
+          python3 third_party/ascend/examples/tutorials/11-rab_time.py
+          #python3 third_party/ascend/examples/tutorials/12-hstu_attention.py
+          python3 third_party/ascend/examples/tutorials/13-matrix-multiplication-optimized-flagtree.py
+          python3 third_party/ascend/examples/tutorials/13-matrix-multiplication-optimized.py
+          python3 third_party/ascend/examples/tutorials/14-accuracy-comparison.py