From 253c021e7783fdcdd11813572ae1dc11260f406f Mon Sep 17 00:00:00 2001 From: doctorpangloss <@hiddenswitch.com> Date: Fri, 13 Sep 2024 09:02:47 -0700 Subject: [PATCH] Add more environment flags for ROCm --- .github/workflows/test.yml | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml index 0907c94bb..13931d53b 100644 --- a/.github/workflows/test.yml +++ b/.github/workflows/test.yml @@ -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