59 lines
1.6 KiB
Nix
59 lines
1.6 KiB
Nix
{ lib
|
|
, stdenv
|
|
, fetchFromGitHub
|
|
, runCommand
|
|
, substituteAll
|
|
, cmake
|
|
}:
|
|
|
|
let
|
|
rocm_version = with lib; concatStrings (intersperse "0" (splitString "." stdenv.cc.version));
|
|
in stdenv.mkDerivation (finalAttrs: {
|
|
pname = "rocm-core";
|
|
version = stdenv.cc.version;
|
|
|
|
# Based on https://github.com/rocm-arch/rocm-arch/tree/ad0b15690d403e5822db062ffff4db3912de6669/rocm-core
|
|
src = let
|
|
rocm_major = lib.versions.major finalAttrs.version;
|
|
rocm_minor = lib.versions.minor finalAttrs.version;
|
|
rocm_patch = lib.versions.patch finalAttrs.version;
|
|
|
|
cmake_lists = substituteAll {
|
|
inherit rocm_version;
|
|
src = ./src/CMakeLists.txt;
|
|
};
|
|
|
|
version_c = substituteAll {
|
|
inherit rocm_major rocm_minor rocm_patch;
|
|
src = ./src/rocm_version.c;
|
|
};
|
|
|
|
version_h = substituteAll {
|
|
inherit rocm_major rocm_minor rocm_patch;
|
|
src = ./src/rocm_version.h;
|
|
};
|
|
in runCommand "rocm-core-${finalAttrs.version}-source" { preferLocalBuild = true; } ''
|
|
mkdir -p $out/rocm-core
|
|
ln -s ${cmake_lists} $out/CMakeLists.txt
|
|
ln -s ${version_c} $out/rocm_version.c
|
|
ln -s ${version_h} $out/rocm-core/rocm_version.h
|
|
'';
|
|
|
|
nativeBuildInputs = [ cmake ];
|
|
|
|
postInstall = ''
|
|
mkdir -p $out/include
|
|
cp -a ../rocm-core $out/include
|
|
ln -s $out/include/rocm-core/rocm_version.h $out/include
|
|
ln -s $out/lib/librocm-core.so.1.0.${rocm_version} $out/lib/librocm-core.so.1
|
|
'';
|
|
|
|
meta = with lib; {
|
|
description = "ROCm core";
|
|
homepage = "https://docs.amd.com";
|
|
license = with licenses; [ ncsa ]; # See src/rocm_version.h
|
|
maintainers = teams.rocm.members;
|
|
platforms = platforms.linux;
|
|
};
|
|
})
|