zhangyiss/spack
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

kokkos: add gfx940 support (#41659)

Browse Source
This commit is contained in:
Richard Berger 2024-01-08 14:33:03 -07:00 committed by GitHub
parent aa768938ff
commit adef1460b4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
var/spack/repos/builtin/packages/kokkos/package.py
View File

@ -158,10 +158,10 @@ class Kokkos(CMakePackage, CudaPackage, ROCmPackage):
"gfx906": "vega906", "gfx906": "vega906",
"gfx908": "vega908", "gfx908": "vega908",
"gfx90a": "vega90A", "gfx90a": "vega90A",
"gfx940": "amd_gfx940",
"gfx942": "amd_gfx942", "gfx942": "amd_gfx942",
"gfx1030": "navi1030", "gfx1030": "navi1030",
"gfx1100": "navi1100", "gfx1100": "navi1100",
"gfx942": "amd_gfx942",
} }
amd_support_conflict_msg = ( amd_support_conflict_msg = (
"{0} is not supported; " "{0} is not supported; "
@ -240,6 +240,12 @@ class Kokkos(CMakePackage, CudaPackage, ROCmPackage):
# Patches # Patches
patch("hpx_profiling_fences.patch", when="@3.5.00 +hpx") patch("hpx_profiling_fences.patch", when="@3.5.00 +hpx")
patch("sycl_bhalft_test.patch", when="@4.2.00 +sycl") patch("sycl_bhalft_test.patch", when="@4.2.00 +sycl")
# adds amd_gfx940 support to Kokkos 4.2.00 (upstreamed in https://github.com/kokkos/kokkos/pull/6671)
patch(
"https://github.com/rbberger/kokkos/commit/293319c5844f4d8eea51eb9cd1457115a5016d3f.patch?full_index=1",
sha256="145619e87dbf26b66ea23e76906576e2a854a3b09f2a2dd70363e61419fa6a6e",
when="@4.2.00",
)
variant("shared", default=True, description="Build shared libraries") variant("shared", default=True, description="Build shared libraries")