rocm

ghex-org · Jan 27, 2025 · 5a61072 · 5a61072
1 parent a14d98c
commit 5a61072
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/.github/workflows/hwmalloc.yml b/.github/workflows/hwmalloc.yml
@@ -17,7 +17,7 @@ jobs:
       matrix:
         package-version: ['0.3.0', 'master']
         compiler: ['[email protected]']
-        device: ['cpu', 'cuda']
+        device: ['cpu', 'cuda', 'rocm']
     with:
       package-name: hwmalloc
       package-version: ${{ matrix.package-version }}

diff --git a/.github/workflows/oomph.yml b/.github/workflows/oomph.yml
@@ -30,6 +30,6 @@ jobs:
         %${{ matrix.compiler }}
         backend=${{ matrix.backend }}
         ${{ matrix.fortran-bindings && '+fortran-bindings' || '' }}
-        ${{ matrix.device == 'cuda' && '+cuda' || '' }}
-        ${{ matrix.device == 'rocm' && '+rocm' || '' }}
+        ${{ matrix.device == 'cuda' && '+cuda cuda_arch=90' || '' }}
+        ${{ matrix.device == 'rocm' && '+rocm amgdgpu_target=gfx942' || '' }}
       test-package: ${{ matrix.device == 'cpu' }}