Skip to content
Draft
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
28 changes: 4 additions & 24 deletions .github/workflows/sycl_python_test.yml
Original file line number Diff line number Diff line change
Expand Up @@ -54,27 +54,19 @@ jobs:
env >> $GITHUB_ENV
sycl-ls

- name: Install DPCTL
id: install_dpctl
uses: ./.github/actions/install-dpctl
with:
DPCTL_URL: https://github.com/IntelPython/dpctl.git
DPCTL_BRANCH: master
DPCTL_PATH: ~/dpctl
VENV_PATH: ~/.venv

- name: Install Torch XPU
- name: Install Sycl-TLA & Torch XPU
shell: bash
run: |
source ~/.venv/bin/activate
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/test/xpu
pip uninstall --yes intel-sycl-rt intel-cmplr-lib-ur umf
pip install build setuptools
python -m build
pip install torch torchvision torchaudio dist/*.whl --extra-index-url https://download.pytorch.org/whl/test/xpu

- name: Run Python Interface GEMM Tests
shell: bash
run: |
source ~/.venv/bin/activate
pip install -e .
export CUTLASS_USE_SYCL=1
export SYCL_UR_TRACE=2
export ONEAPI_DEVICE_SELECTOR=level_zero:gpu
Expand All @@ -95,15 +87,3 @@ jobs:
rm -rf build/ || true
echo "DPC++ cleanup completed"

- name: Cleanup DPCTL
if: always()
shell: bash
run: |
echo "Cleaning up DPCTL installation..."
DPCTL_PATH="${{ inputs.DPCTL_PATH || '~/dpctl' }}"
DPCTL_PATH=$(eval echo $DPCTL_PATH) # Expand ~ to home directory
if [ -d "$DPCTL_PATH" ]; then
echo "Removing DPCTL directory: $DPCTL_PATH"
sudo rm -rf "$DPCTL_PATH"
fi
echo "DPCTL cleanup completed"
3 changes: 2 additions & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,8 @@ dependencies = [
"numpy",
"pydot",
"scipy",
"treelib"
"treelib",
"dpctl"
]

[project.urls]
Expand Down
3 changes: 2 additions & 1 deletion python/setup_cutlass.py
Original file line number Diff line number Diff line change
Expand Up @@ -69,6 +69,7 @@
'pybind11',
'scikit-build',
'treelib',
'pydot'
'pydot',
'dpctl'
]
)
Loading