72 lines
2 KiB
Nix
72 lines
2 KiB
Nix
{ lib, stdenv
|
|
, fetchFromGitHub
|
|
, writeScript
|
|
, cmake
|
|
, rocm-cmake
|
|
, clang
|
|
, rocm-comgr
|
|
, rocm-opencl-runtime
|
|
, rocm-runtime
|
|
, rocm-thunk
|
|
, libelf
|
|
, libglvnd
|
|
, libX11
|
|
, numactl
|
|
}:
|
|
|
|
stdenv.mkDerivation rec {
|
|
pname = "rocclr";
|
|
version = "4.3.1";
|
|
|
|
src = fetchFromGitHub {
|
|
owner = "ROCm-Developer-Tools";
|
|
repo = "ROCclr";
|
|
rev = "rocm-${version}";
|
|
hash = "sha256-3lk7Zucoam+11gFBzg/TWQI1L8uAlxTrPz/mDwTwod4=";
|
|
};
|
|
|
|
nativeBuildInputs = [ cmake rocm-cmake ];
|
|
|
|
buildInputs = [ clang rocm-comgr rocm-runtime rocm-thunk ];
|
|
|
|
propagatedBuildInputs = [ libelf libglvnd libX11 numactl ];
|
|
|
|
prePatch = ''
|
|
substituteInPlace CMakeLists.txt \
|
|
--replace 'set (CMAKE_LIBRARY_OUTPUT_DIRECTORY ''${CMAKE_CURRENT_BINARY_DIR}/lib)' \
|
|
'set (CMAKE_LIBRARY_OUTPUT_DIRECTORY ''${CMAKE_INSTALL_LIBDIR})'
|
|
substituteInPlace device/comgrctx.cpp \
|
|
--replace "libamd_comgr.so" "${rocm-comgr}/lib/libamd_comgr.so"
|
|
'';
|
|
|
|
cmakeFlags = [
|
|
"-DOPENCL_DIR=${rocm-opencl-runtime.src}"
|
|
];
|
|
|
|
preFixup = ''
|
|
# Work around broken cmake files
|
|
ln -s $out/include/compiler/lib/include/* $out/include
|
|
ln -s $out/include/elf/elfio $out/include/elfio
|
|
|
|
substituteInPlace $out/lib/cmake/rocclr/ROCclrConfig.cmake \
|
|
--replace "/build/source/build" "$out"
|
|
'';
|
|
|
|
passthru.updateScript = writeScript "update.sh" ''
|
|
#!/usr/bin/env nix-shell
|
|
#!nix-shell -i bash -p curl jq common-updater-scripts
|
|
version="$(curl -sL "https://api.github.com/repos/ROCm-Developer-Tools/ROCclr/tags" | jq '.[].name | split("-") | .[1] | select( . != null )' --raw-output | sort -n | tail -1)"
|
|
update-source-version rocclr "$version"
|
|
'';
|
|
|
|
meta = with lib; {
|
|
description = "Radeon Open Compute common language runtime";
|
|
homepage = "https://github.com/ROCm-Developer-Tools/ROCclr";
|
|
license = licenses.mit;
|
|
maintainers = with maintainers; [ lovesegfault ];
|
|
# rocclr seems to have some AArch64 ifdefs, but does not seem
|
|
# to be supported yet by the build infrastructure. Recheck in
|
|
# the future.
|
|
platforms = [ "x86_64-linux" ];
|
|
};
|
|
}
|