Skip to content

Commit

Permalink
Unify builds for cpu-backend and triton-shared (#56)
Browse files Browse the repository at this point in the history
* Unify cpu-backend and triton-shared builds
  • Loading branch information
nhat-nguyen authored Nov 16, 2023
1 parent c2fb18d commit ea77225
Show file tree
Hide file tree
Showing 3 changed files with 15 additions and 86 deletions.
7 changes: 0 additions & 7 deletions .github/workflows/integration-tests.yml
Original file line number Diff line number Diff line change
Expand Up @@ -13,10 +13,3 @@ jobs:
# [BACKEND] Fix layouts in nvgpu RewriteTensorPointer. (#2634)
triton-ref: '0e3bf3f58061bd62725e7c502e7e0af19690df1c'
triton-shared-ref: ${{ github.ref }}

test-cpuref:
uses: ./.github/workflows/test-cpuref.yml
with:
# [BACKEND] Fix layouts in nvgpu RewriteTensorPointer. (#2634)
triton-ref: '0e3bf3f58061bd62725e7c502e7e0af19690df1c'
triton-shared-ref: ${{ github.ref }}
79 changes: 0 additions & 79 deletions .github/workflows/test-cpuref.yml

This file was deleted.

15 changes: 15 additions & 0 deletions .github/workflows/test-plugin.yml
Original file line number Diff line number Diff line change
Expand Up @@ -82,3 +82,18 @@ jobs:
echo "Coult not find '${LIT_TEST_DIR}'" ; exit -1
fi
lit -v "${LIT_TEST_DIR}"
- name: Install CPU backend example dependencies
run: |
python3 -m pip install torch --index-url https://download.pytorch.org/whl/cpu
python3 -m pip install pytest
- name: Prepare CPU backend environment
working-directory: triton/python
run: |
echo "TRITON_SHARED_OPT_PATH=$(pwd)/build/$(ls $(pwd)/build | grep -i cmake)/third_party/triton_shared/tools/triton-shared-opt/triton-shared-opt" >> "${GITHUB_ENV}"
echo "LLVM_BINARY_DIR=${HOME}/.triton/llvm/$(ls ${HOME}/.triton/llvm/ | grep -i llvm)/bin" >> "${GITHUB_ENV}"
- name: Run CPU backend examples
working-directory: triton/third_party/triton_shared/python/examples
run: pytest .

0 comments on commit ea77225

Please sign in to comment.