blob: b7fa373e80a72f1a5546a481b0040fbb9362346d (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
|
{
fetchFromGitHub,
stdenv,
cmake,
clr,
numactl,
nlohmann_json,
}:
stdenv.mkDerivation {
pname = "mscclpp";
version = "unstable-2024-12-13";
src = fetchFromGitHub {
owner = "microsoft";
repo = "mscclpp";
rev = "ee75caf365a27b9ab7521cfdda220b55429e5c37";
hash = "sha256-/mi9T9T6OIVtJWN3YoEe9az/86rz7BrX537lqaEh3ig=";
};
nativeBuildInputs = [
cmake
];
buildInputs = [
clr
numactl
];
postPatch = ''
substituteInPlace CMakeLists.txt \
--replace-fail "gfx90a gfx941 gfx942" "gfx908 gfx90a gfx942 gfx1030 gfx1100"
'';
cmakeFlags = [
"-DMSCCLPP_BYPASS_GPU_CHECK=ON"
"-DMSCCLPP_USE_ROCM=ON"
"-DMSCCLPP_BUILD_TESTS=OFF"
"-DGPU_TARGETS=gfx908;gfx90a;gfx942;gfx1030;gfx1100"
"-DAMDGPU_TARGETS=gfx908;gfx90a;gfx942;gfx1030;gfx1100"
"-DMSCCLPP_BUILD_APPS_NCCL=ON"
"-DMSCCLPP_BUILD_PYTHON_BINDINGS=OFF"
"-DFETCHCONTENT_QUIET=OFF"
"-DFETCHCONTENT_TRY_FIND_PACKAGE_MODE=ALWAYS"
"-DFETCHCONTENT_SOURCE_DIR_JSON=${nlohmann_json.src}"
];
env.ROCM_PATH = clr;
}
|
