[FRONTEND] interpreter rewrite (#2321)

This is a new interpreter mode that shares semantic analysis with the
JIT'ed codepath and that the Triton core team is committed to maintain
This commit is contained in:
Philippe Tillet
2023-09-17 14:58:50 -07:00
committed by GitHub
parent 2b066000aa
commit e686b4d6d4
17 changed files with 599 additions and 1033 deletions

View File

@@ -33,6 +33,7 @@ jobs:
echo '::set-output name=matrix-optional::["ubuntu-latest"]'
fi
Integration-Tests-Nvidia:
needs: Runner-Preparation
@@ -119,6 +120,14 @@ jobs:
run: |
rm -rf ~/.triton
- name: Run interpreter tests
env:
# TRITON_INTERPRET: "1"
CUA_VISIBLE_DEVICES: ""
run: |
cd python/test/unit
python3 -m pytest -vs operators/test_flash_attention.py
- name: Run partial tests on CUDA with ENABLE_TMA=1 and ENABLE_MMA_V3=1
if: ${{ env.BACKEND == 'CUDA' && env.ENABLE_TMA == '1' && env.ENABLE_MMA_V3 == '1'}}
run: |