Skip to content

Commit d08f444

Browse files
committed
ROCm UMA
1 parent e072b20 commit d08f444

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

.devops/nix/package.nix

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -173,6 +173,7 @@ effectiveStdenv.mkDerivation (finalAttrs: {
173173
(cmakeBool "GGML_BLAS" useBlas)
174174
(cmakeBool "GGML_CUDA" useCuda)
175175
(cmakeBool "GGML_HIP" useRocm)
176+
(cmakeBool "GGML_HIP_UMA" useRocm)
176177
(cmakeBool "GGML_METAL" useMetalKit)
177178
(cmakeBool "GGML_VULKAN" useVulkan)
178179
(cmakeBool "GGML_STATIC" enableStatic)
@@ -189,6 +190,7 @@ effectiveStdenv.mkDerivation (finalAttrs: {
189190
++ optionals useRocm [
190191
(cmakeFeature "CMAKE_HIP_COMPILER" "${rocmPackages.llvm.clang}/bin/clang")
191192
(cmakeFeature "CMAKE_HIP_ARCHITECTURES" rocmGpuTargets)
193+
(cmakeFeature "AMDGPU_TARGETS" "gfx1030")
192194
]
193195
++ optionals useMetalKit [
194196
(lib.cmakeFeature "CMAKE_C_FLAGS" "-D__ARM_FEATURE_DOTPROD=1")

0 commit comments

Comments
 (0)