mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-01-12 15:20:51 +08:00
Add more environment flags for ROCm
This commit is contained in:
parent
5d1eda7832
commit
253c021e77
3
.github/workflows/test.yml
vendored
3
.github/workflows/test.yml
vendored
@ -73,7 +73,8 @@ jobs:
|
||||
pip install --break-system-packages --no-build-isolation .[dev] || pip install --no-build-isolation .[dev]
|
||||
- name: Run tests
|
||||
run: |
|
||||
export HSA_OVERRIDE_GFX_VERSION=11.0.2
|
||||
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
|
||||
|
||||
Loading…
Reference in New Issue
Block a user