Luna Nova
68 lines
1.3 KiB
Nix
68 lines
1.3 KiB
Nix
{
|
|
lib,
|
|
stdenv,
|
|
cmake,
|
|
ninja,
|
|
zlib,
|
|
zstd,
|
|
llvm,
|
|
python3,
|
|
}:
|
|
|
|
let
|
|
llvmNativeTarget =
|
|
if stdenv.hostPlatform.isx86_64 then
|
|
"X86"
|
|
else if stdenv.hostPlatform.isAarch64 then
|
|
"AArch64"
|
|
else
|
|
throw "Unsupported ROCm LLVM platform";
|
|
in
|
|
stdenv.mkDerivation (finalAttrs: {
|
|
pname = "rocm-device-libs";
|
|
# In-tree with ROCm LLVM
|
|
inherit (llvm.llvm) version;
|
|
src = llvm.llvm.monorepoSrc;
|
|
sourceRoot = "${finalAttrs.src.name}/amd/device-libs";
|
|
strictDeps = true;
|
|
__structuredAttrs = true;
|
|
|
|
postPatch =
|
|
# Use our sysrooted toolchain instead of direct clang target
|
|
''
|
|
substituteInPlace cmake/OCL.cmake \
|
|
--replace-fail '$<TARGET_FILE:clang>' "${llvm.rocm-toolchain}/bin/clang"
|
|
'';
|
|
|
|
patches = [
|
|
./cmake.patch
|
|
];
|
|
|
|
nativeBuildInputs = [
|
|
cmake
|
|
ninja
|
|
python3
|
|
llvm.rocm-toolchain
|
|
];
|
|
|
|
buildInputs = [
|
|
llvm.llvm
|
|
llvm.clang-unwrapped
|
|
zlib
|
|
zstd
|
|
];
|
|
|
|
cmakeFlags = [
|
|
"-DLLVM_TARGETS_TO_BUILD=AMDGPU;${llvmNativeTarget}"
|
|
];
|
|
|
|
meta = {
|
|
description = "Set of AMD-specific device-side language runtime libraries";
|
|
homepage = "https://github.com/ROCm/ROCm-Device-Libs";
|
|
license = lib.licenses.ncsa;
|
|
maintainers = with lib.maintainers; [ lovesegfault ];
|
|
teams = [ lib.teams.rocm ];
|
|
platforms = lib.platforms.linux;
|
|
};
|
|
})
|