mirror of
https://github.com/NixOS/nixpkgs.git
synced 2024-12-15 18:23:09 +00:00
3ced8179cd
hip: optional GITHUB_TOKEN for update script rocm-llvm: optional GITHUB_TOKEN for update script clang-ocl: optional GITHUB_TOKEN for update script hipcub: optional GITHUB_TOKEN for update script hipsparse: optional GITHUB_TOKEN for update script miopen: optional GITHUB_TOKEN for update script miopengemm: optional GITHUB_TOKEN for update script rccl: optional GITHUB_TOKEN for update script rocblas: optional GITHUB_TOKEN for update script rocclr: optional GITHUB_TOKEN for update script rocfft: optional GITHUB_TOKEN for update script rocm-comgr: optional GITHUB_TOKEN for update script rocm-device-libs: optional GITHUB_TOKEN for update script rocm-opencl-runtime: optional GITHUB_TOKEN for update script rocm-runtime: optional GITHUB_TOKEN for update script rocm-thunk: optional GITHUB_TOKEN for update script rocmlir: optional GITHUB_TOKEN for update script rocprim: optional GITHUB_TOKEN for update script rocrand: optional GITHUB_TOKEN for update script rocsparse: optional GITHUB_TOKEN for update script rocthrust: optional GITHUB_TOKEN for update script rocwmma: optional GITHUB_TOKEN for update script tensile: optional GITHUB_TOKEN for update script rocm-cmake: optional GITHUB_TOKEN for update script rocminfo: optional GITHUB_TOKEN for update script rocm-smi: optional GITHUB_TOKEN for update script
207 lines
6.9 KiB
Nix
207 lines
6.9 KiB
Nix
{ stdenv
|
|
, binutils-unwrapped
|
|
, clang
|
|
, cmake
|
|
, fetchFromGitHub
|
|
, fetchpatch
|
|
, file
|
|
, lib
|
|
, libglvnd
|
|
, libX11
|
|
, libxml2
|
|
, llvm
|
|
, makeWrapper
|
|
, numactl
|
|
, perl
|
|
, python3
|
|
, python3Packages
|
|
, rocclr
|
|
, rocm-comgr
|
|
, rocm-device-libs
|
|
, rocm-opencl-runtime
|
|
, rocm-runtime
|
|
, rocm-thunk
|
|
, rocminfo
|
|
, substituteAll
|
|
, writeScript
|
|
, writeText
|
|
}:
|
|
|
|
let
|
|
hip = stdenv.mkDerivation (finalAttrs: {
|
|
pname = "hip";
|
|
version = "5.3.3";
|
|
|
|
src = fetchFromGitHub {
|
|
owner = "ROCm-Developer-Tools";
|
|
repo = "HIP";
|
|
rev = "rocm-${finalAttrs.version}";
|
|
hash = "sha256-kmRvrwnT0h2dBMI+H9d1vmeW3TmDBD+qW4YYhaMV2dE=";
|
|
};
|
|
|
|
patches = [
|
|
(substituteAll {
|
|
src = ./hip-config-paths.patch;
|
|
inherit llvm;
|
|
rocm_runtime = rocm-runtime;
|
|
})
|
|
];
|
|
|
|
# - fix bash paths
|
|
# - fix path to rocm_agent_enumerator
|
|
# - fix hcc path
|
|
# - fix hcc version parsing
|
|
# - add linker flags for libhsa-runtime64 and hc_am since libhip_hcc
|
|
# refers to them.
|
|
prePatch = ''
|
|
for f in $(find bin -type f); do
|
|
sed -e 's,#!/usr/bin/perl,#!${perl}/bin/perl,' \
|
|
-e 's,#!/bin/bash,#!${stdenv.shell},' \
|
|
-i "$f"
|
|
done
|
|
|
|
substituteInPlace bin/hip_embed_pch.sh \
|
|
--replace '$LLVM_DIR/bin/' ""
|
|
|
|
sed 's,#!/usr/bin/python,#!${python3.interpreter},' -i hip_prof_gen.py
|
|
|
|
sed -e 's,$ROCM_AGENT_ENUM = "''${ROCM_PATH}/bin/rocm_agent_enumerator";,$ROCM_AGENT_ENUM = "${rocminfo}/bin/rocm_agent_enumerator";,' \
|
|
-e 's,^\($DEVICE_LIB_PATH=\).*$,\1"${rocm-device-libs}/amdgcn/bitcode";,' \
|
|
-e 's,^\($HIP_COMPILER=\).*$,\1"clang";,' \
|
|
-e 's,^\($HIP_RUNTIME=\).*$,\1"ROCclr";,' \
|
|
-e 's,^\([[:space:]]*$HSA_PATH=\).*$,\1"${rocm-runtime}";,'g \
|
|
-e 's,^\([[:space:]]*\)$HIP_CLANG_INCLUDE_PATH = abs_path("$HIP_CLANG_PATH/../lib/clang/$HIP_CLANG_VERSION/include");,\1$HIP_CLANG_INCLUDE_PATH = "${llvm}/lib/clang/$HIP_CLANG_VERSION/include";,' \
|
|
-e 's,^\([[:space:]]*$HIPCXXFLAGS .= " -isystem \\"$HIP_CLANG_INCLUDE_PATH/..\\"\)";,\1 -isystem ${rocm-runtime}/include";,' \
|
|
-e 's,$HIP_CLANG_PATH/../lib/clang/$HIP_CLANG_VERSION,$HIP_CLANG_PATH/../resource-root,g' \
|
|
-e 's,`file,`${file}/bin/file,g' \
|
|
-e 's,`readelf,`${binutils-unwrapped}/bin/readelf,' \
|
|
-e 's, ar , ${binutils-unwrapped}/bin/ar ,g' \
|
|
-i bin/hipcc.pl
|
|
|
|
sed -e 's,^\($HSA_PATH=\).*$,\1"${rocm-runtime}";,' \
|
|
-e 's,^\($HIP_CLANG_PATH=\).*$,\1"${clang}/bin";,' \
|
|
-e 's,^\($HIP_PLATFORM=\).*$,\1"amd";,' \
|
|
-e 's,$HIP_CLANG_PATH/llc,${llvm}/bin/llc,' \
|
|
-e 's, abs_path, Cwd::abs_path,' \
|
|
-i bin/hipconfig.pl
|
|
|
|
sed -e 's, abs_path, Cwd::abs_path,' -i bin/hipvars.pm
|
|
'';
|
|
|
|
buildPhase = "";
|
|
|
|
installPhase = ''
|
|
runHook preInstall
|
|
|
|
mkdir -p $out
|
|
cp -r * $out/
|
|
|
|
runHook postInstall
|
|
'';
|
|
|
|
meta = with lib; {
|
|
description = "C++ Heterogeneous-Compute Interface for Portability";
|
|
homepage = "https://github.com/ROCm-Developer-Tools/HIP";
|
|
license = licenses.mit;
|
|
maintainers = with maintainers; [ lovesegfault ] ++ teams.rocm.members;
|
|
platforms = platforms.linux;
|
|
};
|
|
});
|
|
in
|
|
stdenv.mkDerivation (finalAttrs: {
|
|
pname = "hip";
|
|
version = "5.3.3";
|
|
|
|
src = fetchFromGitHub {
|
|
owner = "ROCm-Developer-Tools";
|
|
repo = "hipamd";
|
|
rev = "rocm-${finalAttrs.version}";
|
|
hash = "sha256-i7hT/j+V0LT6Va2XcQyyKXF1guoIyhcOHvn842wCRx4=";
|
|
};
|
|
|
|
nativeBuildInputs = [ cmake python3 makeWrapper perl ];
|
|
buildInputs = [ libxml2 numactl libglvnd libX11 python3Packages.cppheaderparser ];
|
|
propagatedBuildInputs = [
|
|
clang
|
|
llvm
|
|
rocm-comgr
|
|
rocm-device-libs
|
|
rocm-runtime
|
|
rocm-thunk
|
|
rocminfo
|
|
];
|
|
|
|
patches = [
|
|
(substituteAll {
|
|
src = ./hipamd-config-paths.patch;
|
|
inherit clang llvm hip;
|
|
rocm_runtime = rocm-runtime;
|
|
})
|
|
];
|
|
|
|
prePatch = ''
|
|
sed -e 's,#!/bin/bash,#!${stdenv.shell},' \
|
|
-i src/hip_embed_pch.sh
|
|
'';
|
|
|
|
preConfigure = ''
|
|
export HIP_CLANG_PATH=${clang}/bin
|
|
export DEVICE_LIB_PATH=${rocm-device-libs}/lib
|
|
'';
|
|
|
|
cmakeFlags = [
|
|
"-DHIP_PLATFORM=amd"
|
|
"-DAMD_OPENCL_PATH=${rocm-opencl-runtime.src}"
|
|
"-DHIP_COMMON_DIR=${hip}"
|
|
"-DROCCLR_PATH=${rocclr}"
|
|
"-DHIP_VERSION_BUILD_ID=0"
|
|
# Temporarily set variables to work around upstream CMakeLists issue
|
|
# Can be removed once https://github.com/ROCm-Developer-Tools/hipamd/issues/55 is fixed
|
|
"-DCMAKE_INSTALL_BINDIR=bin"
|
|
"-DCMAKE_INSTALL_INCLUDEDIR=include"
|
|
"-DCMAKE_INSTALL_LIBDIR=lib"
|
|
];
|
|
|
|
postInstall = ''
|
|
patchShebangs $out/bin
|
|
wrapProgram $out/bin/hipcc --set HIP_PATH $out --set HSA_PATH ${rocm-runtime} --set HIP_CLANG_PATH ${clang}/bin --prefix PATH : ${llvm}/bin --set ROCM_PATH $out
|
|
wrapProgram $out/bin/hipconfig --set HIP_PATH $out --set HSA_PATH ${rocm-runtime} --set HIP_CLANG_PATH ${clang}/bin
|
|
'';
|
|
|
|
passthru.updateScript = writeScript "update.sh" ''
|
|
#!/usr/bin/env nix-shell
|
|
#!nix-shell -i bash -p curl jq common-updater-scripts nix-prefetch-github
|
|
version="$(curl ''${GITHUB_TOKEN:+"-u \":$GITHUB_TOKEN\""} -sL "https://api.github.com/repos/ROCm-Developer-Tools/HIP/tags" | jq '.[].name | split("-") | .[1] | select( . != null )' --raw-output | sort -n | tail -1)"
|
|
current_version="$(grep "version =" pkgs/development/compilers/hip/default.nix | head -n1 | cut -d'"' -f2)"
|
|
if [[ "$version" != "$current_version" ]]; then
|
|
tarball_meta="$(nix-prefetch-github ROCm-Developer-Tools HIP --rev "rocm-$version")"
|
|
tarball_hash="$(nix to-base64 sha256-$(jq -r '.sha256' <<< "$tarball_meta"))"
|
|
sed -i -z "pkgs/development/compilers/hip/default.nix" \
|
|
-e 's,version = "[^'"'"'"]*",version = "'"$version"'",1' \
|
|
-e 's,hash = "[^'"'"'"]*",hash = "sha256-'"$tarball_hash"'",1'
|
|
else
|
|
echo hip already up-to-date
|
|
fi
|
|
|
|
version="$(curl ''${GITHUB_TOKEN:+"-u \":$GITHUB_TOKEN\""} -sL "https://api.github.com/repos/ROCm-Developer-Tools/hipamd/tags" | jq '.[].name | split("-") | .[1] | select( . != null )' --raw-output | sort -n | tail -1)"
|
|
current_version="$(grep "version =" pkgs/development/compilers/hip/default.nix | tail -n1 | cut -d'"' -f2)"
|
|
if [[ "$version" != "$current_version" ]]; then
|
|
tarball_meta="$(nix-prefetch-github ROCm-Developer-Tools hipamd --rev "rocm-$version")"
|
|
tarball_hash="$(nix to-base64 sha256-$(jq -r '.sha256' <<< "$tarball_meta"))"
|
|
sed -i -z "pkgs/development/compilers/hip/default.nix" \
|
|
-e 's,version = "[^'"'"'"]*",version = "'"$version"'",2' \
|
|
-e 's,hash = "[^'"'"'"]*",hash = "sha256-'"$tarball_hash"'",2'
|
|
else
|
|
echo hipamd already up-to-date
|
|
fi
|
|
'';
|
|
|
|
meta = with lib; {
|
|
description = "C++ Heterogeneous-Compute Interface for Portability";
|
|
homepage = "https://github.com/ROCm-Developer-Tools/hipamd";
|
|
license = licenses.mit;
|
|
maintainers = with maintainers; [ lovesegfault ] ++ teams.rocm.members;
|
|
platforms = platforms.linux;
|
|
};
|
|
})
|