mirror of
https://github.com/NixOS/nixpkgs.git
synced 2025-01-07 21:43:32 +00:00
4f0dadbf38
After final improvements to the official formatter implementation,
this commit now performs the first treewide reformat of Nix files using it.
This is part of the implementation of RFC 166.
Only "inactive" files are reformatted, meaning only files that
aren't being touched by any PR with activity in the past 2 months.
This is to avoid conflicts for PRs that might soon be merged.
Later we can do a full treewide reformat to get the rest,
which should not cause as many conflicts.
A CI check has already been running for some time to ensure that new and
already-formatted files are formatted, so the files being reformatted here
should also stay formatted.
This commit was automatically created and can be verified using
nix-build a08b3a4d19.tar.gz \
--argstr baseRev b32a094368
result/bin/apply-formatting $NIXPKGS_PATH
188 lines
4.1 KiB
Nix
188 lines
4.1 KiB
Nix
{
|
|
lib,
|
|
stdenv,
|
|
fetchFromGitHub,
|
|
rocmUpdateScript,
|
|
pkg-config,
|
|
cmake,
|
|
rocm-cmake,
|
|
clr,
|
|
clang-tools-extra,
|
|
openmp,
|
|
rocblas,
|
|
rocmlir,
|
|
composable_kernel,
|
|
miopen,
|
|
protobuf,
|
|
half,
|
|
nlohmann_json,
|
|
msgpack,
|
|
sqlite,
|
|
oneDNN_2,
|
|
blaze,
|
|
cppcheck,
|
|
rocm-device-libs,
|
|
texliveSmall,
|
|
doxygen,
|
|
sphinx,
|
|
docutils,
|
|
ghostscript,
|
|
python3Packages,
|
|
buildDocs ? false,
|
|
buildTests ? false,
|
|
gpuTargets ? clr.gpuTargets,
|
|
}:
|
|
|
|
let
|
|
latex = lib.optionalAttrs buildDocs (
|
|
texliveSmall.withPackages (
|
|
ps: with ps; [
|
|
latexmk
|
|
tex-gyre
|
|
fncychap
|
|
wrapfig
|
|
capt-of
|
|
framed
|
|
needspace
|
|
tabulary
|
|
varwidth
|
|
titlesec
|
|
epstopdf
|
|
]
|
|
)
|
|
);
|
|
in
|
|
stdenv.mkDerivation (finalAttrs: {
|
|
pname = "migraphx";
|
|
version = "6.0.2";
|
|
|
|
outputs =
|
|
[
|
|
"out"
|
|
]
|
|
++ lib.optionals buildDocs [
|
|
"doc"
|
|
]
|
|
++ lib.optionals buildTests [
|
|
"test"
|
|
];
|
|
|
|
src = fetchFromGitHub {
|
|
owner = "ROCm";
|
|
repo = "AMDMIGraphX";
|
|
rev = "rocm-${finalAttrs.version}";
|
|
hash = "sha256-VDYUSpWYAdJ63SKVCO26DVAC3RtZM7otqN0sYUA6DBQ=";
|
|
};
|
|
|
|
nativeBuildInputs =
|
|
[
|
|
pkg-config
|
|
cmake
|
|
rocm-cmake
|
|
clr
|
|
clang-tools-extra
|
|
python3Packages.python
|
|
]
|
|
++ lib.optionals buildDocs [
|
|
latex
|
|
doxygen
|
|
sphinx
|
|
docutils
|
|
ghostscript
|
|
python3Packages.sphinx-rtd-theme
|
|
python3Packages.breathe
|
|
];
|
|
|
|
buildInputs = [
|
|
openmp
|
|
rocblas
|
|
rocmlir
|
|
composable_kernel
|
|
miopen
|
|
protobuf
|
|
half
|
|
nlohmann_json
|
|
msgpack
|
|
sqlite
|
|
oneDNN_2
|
|
blaze
|
|
cppcheck
|
|
python3Packages.pybind11
|
|
python3Packages.onnx
|
|
];
|
|
|
|
cmakeFlags = [
|
|
"-DMIGRAPHX_ENABLE_GPU=ON"
|
|
"-DMIGRAPHX_ENABLE_CPU=ON"
|
|
"-DMIGRAPHX_ENABLE_FPGA=ON"
|
|
"-DMIGRAPHX_ENABLE_MLIR=OFF" # LLVM or rocMLIR mismatch?
|
|
# Manually define CMAKE_INSTALL_<DIR>
|
|
# See: https://github.com/NixOS/nixpkgs/pull/197838
|
|
"-DCMAKE_INSTALL_BINDIR=bin"
|
|
"-DCMAKE_INSTALL_LIBDIR=lib"
|
|
"-DCMAKE_INSTALL_INCLUDEDIR=include"
|
|
"-DGPU_TARGETS=${lib.concatStringsSep ";" gpuTargets}"
|
|
];
|
|
|
|
postPatch =
|
|
''
|
|
# We need to not use hipcc and define the CXXFLAGS manually due to `undefined hidden symbol: tensorflow:: ...`
|
|
export CXXFLAGS+="--rocm-path=${clr} --rocm-device-lib-path=${rocm-device-libs}/amdgcn/bitcode"
|
|
patchShebangs tools
|
|
|
|
# `error: '__clang_hip_runtime_wrapper.h' file not found [clang-diagnostic-error]`
|
|
substituteInPlace CMakeLists.txt \
|
|
--replace "set(MIGRAPHX_TIDY_ERRORS ALL)" ""
|
|
|
|
# JIT library was removed from composable_kernel...
|
|
# https://github.com/ROCm/composable_kernel/issues/782
|
|
substituteInPlace src/targets/gpu/CMakeLists.txt \
|
|
--replace " COMPONENTS jit_library" "" \
|
|
--replace " composable_kernel::jit_library" "" \
|
|
--replace "if(WIN32)" "if(TRUE)"
|
|
''
|
|
+ lib.optionalString (!buildDocs) ''
|
|
substituteInPlace CMakeLists.txt \
|
|
--replace "add_subdirectory(doc)" ""
|
|
''
|
|
+ lib.optionalString (!buildTests) ''
|
|
substituteInPlace CMakeLists.txt \
|
|
--replace "add_subdirectory(test)" ""
|
|
'';
|
|
|
|
# Unfortunately, it seems like we have to call make on this manually
|
|
preInstall = lib.optionalString buildDocs ''
|
|
export HOME=$(mktemp -d)
|
|
make -j$NIX_BUILD_CORES doc
|
|
cd ../doc/pdf
|
|
make -j$NIX_BUILD_CORES
|
|
cd -
|
|
'';
|
|
|
|
postInstall =
|
|
lib.optionalString buildDocs ''
|
|
mv ../doc/html $out/share/doc/migraphx
|
|
mv ../doc/pdf/MIGraphX.pdf $out/share/doc/migraphx
|
|
''
|
|
+ lib.optionalString buildTests ''
|
|
mkdir -p $test/bin
|
|
mv bin/test_* $test/bin
|
|
patchelf $test/bin/test_* --shrink-rpath --allowed-rpath-prefixes "$NIX_STORE"
|
|
'';
|
|
|
|
passthru.updateScript = rocmUpdateScript {
|
|
name = finalAttrs.pname;
|
|
owner = finalAttrs.src.owner;
|
|
repo = finalAttrs.src.repo;
|
|
};
|
|
|
|
meta = with lib; {
|
|
description = "AMD's graph optimization engine";
|
|
homepage = "https://github.com/ROCm/AMDMIGraphX";
|
|
license = with licenses; [ mit ];
|
|
maintainers = teams.rocm.members;
|
|
platforms = platforms.linux;
|
|
broken = true;
|
|
};
|
|
})
|