Add iree-hip-target flag

Signed-off-by: aviator19941 <[email protected]>
nod-ai · Oct 29, 2024 · 5e1fd8a · 5e1fd8a
1 parent 5085736
commit 5e1fd8a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/ci-llama.yaml b/.github/workflows/ci-llama.yaml
@@ -76,7 +76,7 @@ jobs:
             "numpy<2.0"
 
       - name: Run llama test
-        run: pytest sharktank/tests/models/llama/benchmark_amdgpu_test.py -v -s --longrun
+        run: pytest sharktank/tests/models/llama/benchmark_amdgpu_test.py -v -s --longrun --iree-hip-target=gfx942
 
       - name: Upload llama executable files
         uses: actions/upload-artifact@v4