74 lines
1.7 KiB
Nix
74 lines
1.7 KiB
Nix
|
{
|
||
|
|
lib,
|
||
|
|
stdenv,
|
||
|
|
fetchpatch,
|
||
|
|
cmake,
|
||
|
|
python3,
|
||
|
|
rocm-merged-llvm,
|
||
|
|
rocm-device-libs,
|
||
|
|
zlib,
|
||
|
|
zstd,
|
||
|
|
libxml2,
|
||
|
|
}:
|
||
|
|
|
||
|
|
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-comgr";
|
||
|
|
# In-tree with ROCm LLVM
|
||
|
|
inherit (rocm-merged-llvm) version;
|
||
|
|
src = rocm-merged-llvm.llvm-src;
|
||
|
|
|
||
|
|
sourceRoot = "${finalAttrs.src.name}/amd/comgr";
|
||
|
|
|
||
|
|
patches = [
|
||
|
|
# [Comgr] Extend ISA compatibility
|
||
|
|
(fetchpatch {
|
||
|
|
sha256 = "sha256-dgow0kwSWM1TnkqWOZDRQrh5nuF8p5jbYyOLCpQsH4k=";
|
||
|
|
url = "https://github.com/GZGavinZhao/rocm-llvm-project/commit/a439e4f37ce71de48d4a979594276e3be0e6278f.patch";
|
||
|
|
relative = "amd/comgr";
|
||
|
|
})
|
||
|
|
#[Comgr] Extend ISA compatibility for CCOB
|
||
|
|
(fetchpatch {
|
||
|
|
sha256 = "sha256-PCi0QHLiEQCTIYRtSSbhOjXANJ3zC3VLdMED1BEfQeg=";
|
||
|
|
url = "https://github.com/GZGavinZhao/rocm-llvm-project/commit/fa80abb77d5ae6f8d89ab956e7ebda9c802a804f.patch";
|
||
|
|
relative = "amd/comgr";
|
||
|
|
})
|
||
|
|
];
|
||
|
|
|
||
|
|
nativeBuildInputs = [
|
||
|
|
cmake
|
||
|
|
python3
|
||
|
|
];
|
||
|
|
|
||
|
|
buildInputs = [
|
||
|
|
rocm-device-libs
|
||
|
|
libxml2
|
||
|
|
zlib
|
||
|
|
zstd
|
||
|
|
rocm-merged-llvm
|
||
|
|
];
|
||
|
|
|
||
|
|
cmakeFlags = [
|
||
|
|
"-DCMAKE_VERBOSE_MAKEFILE=ON"
|
||
|
|
"-DCMAKE_BUILD_TYPE=Release"
|
||
|
|
"-DLLVM_TARGETS_TO_BUILD=AMDGPU;${llvmNativeTarget}"
|
||
|
|
];
|
||
|
|
|
||
|
|
meta = with lib; {
|
||
|
|
description = "APIs for compiling and inspecting AMDGPU code objects";
|
||
|
|
homepage = "https://github.com/ROCm/ROCm-CompilerSupport/tree/amd-stg-open/lib/comgr";
|
||
|
|
license = licenses.ncsa;
|
||
|
|
maintainers = with maintainers; [ lovesegfault ];
|
||
|
|
teams = [ teams.rocm ];
|
||
|
|
platforms = platforms.linux;
|
||
|
|
};
|
||
|
|
})
|