wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-11 23:00:51 +08:00
Code Issues Actions 9 Packages Projects Releases Wiki Activity

Re-enable ROCm tests

Browse Source
This commit is contained in:
Benjamin Berman 2024-10-17 11:16:17 -07:00
parent a83b561ea7
commit 7587e42d8b
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
.github/workflows/test.yml vendored
View File

@ -54,6 +54,8 @@ jobs:
container: "ubuntu:22.04"
- labels: [self-hosted, Linux, X64, cuda-3060-12gb]
container: "nvcr.io/nvidia/pytorch:24.03-py3"
- labels: [self-hosted, Linux, X64, rocm-7600-8gb]
container: "rocm/pytorch:rocm6.2.3_ubuntu22.04_py3.10_pytorch_release_2.3.0"
steps:
- run: |
apt update
@ -72,11 +74,8 @@ jobs:
- name: Run tests
run: |
export HSA_OVERRIDE_GFX_VERSION=11.0.0
export TORCH_BLAS_PREFER_HIPBLASLT=0
export HIP_VISIBLE_DEVICES=0
export PYTORCH_HIP_ALLOC_CONF=expandable_segments:True
export NUMBA_THREADING_LAYER=omp
export AMD_SERIALIZE_KERNEL=1
nvidia-smi || rocminfo || true
pytest -v tests/unit
- name: Lint for errors
run: |