2021-01-21 17:00:13 +00:00
|
|
|
{ lib, stdenv
|
2020-07-10 18:34:25 +00:00
|
|
|
, fetchFromGitHub
|
|
|
|
, cmake
|
|
|
|
, rocm-cmake
|
|
|
|
, clang
|
|
|
|
, rocm-comgr
|
|
|
|
, rocm-opencl-runtime
|
|
|
|
, rocm-runtime
|
|
|
|
, rocm-thunk
|
|
|
|
, libelf
|
|
|
|
, libglvnd
|
|
|
|
, libX11
|
2020-08-21 06:38:43 +00:00
|
|
|
, numactl
|
2020-07-10 18:34:25 +00:00
|
|
|
}:
|
|
|
|
|
|
|
|
stdenv.mkDerivation rec {
|
|
|
|
pname = "rocclr";
|
2020-12-18 07:31:49 +00:00
|
|
|
version = "4.0.0";
|
2020-07-10 18:34:25 +00:00
|
|
|
|
|
|
|
src = fetchFromGitHub {
|
|
|
|
owner = "ROCm-Developer-Tools";
|
|
|
|
repo = "ROCclr";
|
2020-08-21 06:38:43 +00:00
|
|
|
rev = "rocm-${version}";
|
2020-12-18 07:31:49 +00:00
|
|
|
hash = "sha256-B27ff1b9JRhxFUsBt7CGuYaR87hvKbVSCERWD45d8tM=";
|
2020-07-10 18:34:25 +00:00
|
|
|
};
|
|
|
|
|
|
|
|
nativeBuildInputs = [ cmake rocm-cmake ];
|
|
|
|
|
2020-08-07 14:49:18 +00:00
|
|
|
buildInputs = [ clang rocm-comgr rocm-runtime rocm-thunk ];
|
2020-07-10 18:34:25 +00:00
|
|
|
|
2020-08-21 06:38:43 +00:00
|
|
|
propagatedBuildInputs = [ libelf libglvnd libX11 numactl ];
|
2020-07-10 18:34:25 +00:00
|
|
|
|
|
|
|
prePatch = ''
|
|
|
|
substituteInPlace CMakeLists.txt \
|
|
|
|
--replace 'set (CMAKE_LIBRARY_OUTPUT_DIRECTORY ''${CMAKE_CURRENT_BINARY_DIR}/lib)' \
|
2021-02-01 11:38:22 +00:00
|
|
|
'set (CMAKE_LIBRARY_OUTPUT_DIRECTORY ''${CMAKE_INSTALL_LIBDIR})'
|
2020-07-10 18:34:25 +00:00
|
|
|
substituteInPlace device/comgrctx.cpp \
|
|
|
|
--replace "libamd_comgr.so" "${rocm-comgr}/lib/libamd_comgr.so"
|
|
|
|
'';
|
|
|
|
|
|
|
|
cmakeFlags = [
|
|
|
|
"-DOPENCL_DIR=${rocm-opencl-runtime.src}"
|
|
|
|
];
|
|
|
|
|
|
|
|
preFixup = ''
|
2020-11-03 07:50:47 +00:00
|
|
|
# Work around broken cmake files
|
2020-07-10 18:34:25 +00:00
|
|
|
ln -s $out/include/compiler/lib/include/* $out/include
|
2020-11-03 07:50:47 +00:00
|
|
|
ln -s $out/include/elf/elfio $out/include/elfio
|
|
|
|
|
2020-08-21 06:38:43 +00:00
|
|
|
substituteInPlace $out/lib/cmake/rocclr/ROCclrConfig.cmake \
|
|
|
|
--replace "/build/source/build" "$out"
|
2020-07-10 18:34:25 +00:00
|
|
|
'';
|
|
|
|
|
2021-01-21 17:00:13 +00:00
|
|
|
meta = with lib; {
|
2020-07-10 18:34:25 +00:00
|
|
|
description = "Radeon Open Compute common language runtime";
|
|
|
|
homepage = "https://github.com/ROCm-Developer-Tools/ROCclr";
|
|
|
|
license = licenses.mit;
|
|
|
|
maintainers = with maintainers; [ danieldk ];
|
|
|
|
# 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" ];
|
|
|
|
};
|
|
|
|
}
|