forked from rarias/jungle
Compare commits
No commits in common. "8f07c3803c0c7e311430e7f0c97b7959ddf8cd2f" and "f5451b3b91e3af043cf9064a338bb9eabce4ca0a" have entirely different histories.
8f07c3803c
...
f5451b3b91
13
overlay.nix
13
overlay.nix
@ -18,12 +18,8 @@ let
|
|||||||
cudainfo = prev.callPackage ./pkgs/cudainfo/default.nix { };
|
cudainfo = prev.callPackage ./pkgs/cudainfo/default.nix { };
|
||||||
#extrae = callPackage ./pkgs/extrae/default.nix { }; # Broken and outdated
|
#extrae = callPackage ./pkgs/extrae/default.nix { }; # Broken and outdated
|
||||||
gpi-2 = callPackage ./pkgs/gpi-2/default.nix { };
|
gpi-2 = callPackage ./pkgs/gpi-2/default.nix { };
|
||||||
intel-apt = callPackage ./pkgs/intel-oneapi/packages.nix { };
|
|
||||||
intelPackages_2023 = callPackage ./pkgs/intel-oneapi/2023.nix { };
|
intelPackages_2023 = callPackage ./pkgs/intel-oneapi/2023.nix { };
|
||||||
intelPackages_2024 = final.intel-apt.hpckit_2024;
|
intelPackages = final.intelPackages_2023;
|
||||||
intelPackages_2025 = final.intel-apt.hpckit_2025;
|
|
||||||
intelPackages = final.intelPackages_2025;
|
|
||||||
oneMath = callPackage ./pkgs/onemath/default.nix { };
|
|
||||||
jemallocNanos6 = callPackage ./pkgs/nanos6/jemalloc.nix { };
|
jemallocNanos6 = callPackage ./pkgs/nanos6/jemalloc.nix { };
|
||||||
# FIXME: Extend this to all linuxPackages variants. Open problem, see:
|
# FIXME: Extend this to all linuxPackages variants. Open problem, see:
|
||||||
# https://discourse.nixos.org/t/whats-the-right-way-to-make-a-custom-kernel-module-available/4636
|
# https://discourse.nixos.org/t/whats-the-right-way-to-make-a-custom-kernel-module-available/4636
|
||||||
@ -84,13 +80,6 @@ let
|
|||||||
intel2023-ifort = callPackage ./test/compilers/hello-f.nix {
|
intel2023-ifort = callPackage ./test/compilers/hello-f.nix {
|
||||||
stdenv = final.intelPackages_2023.stdenv-ifort;
|
stdenv = final.intelPackages_2023.stdenv-ifort;
|
||||||
};
|
};
|
||||||
intel2024-icx-c = hello-c.override { stdenv = final.intelPackages_2024.stdenv; };
|
|
||||||
intel2025-icx-c = hello-c.override { stdenv = final.intelPackages_2025.stdenv; };
|
|
||||||
intel2024-icx-cpp = hello-cpp.override { stdenv = final.intelPackages_2024.stdenv; };
|
|
||||||
intel2025-icx-cpp = hello-cpp.override { stdenv = final.intelPackages_2025.stdenv; };
|
|
||||||
intel2023-sycl = hello-sycl.override { intelPackages = final.intelPackages_2023; };
|
|
||||||
intel2024-sycl = hello-sycl.override { intelPackages = final.intelPackages_2024; };
|
|
||||||
intel2025-sycl = hello-sycl.override { intelPackages = final.intelPackages_2025; };
|
|
||||||
clangOmpss2-lto = lto.override { stdenv = final.stdenvClangOmpss2Nanos6; };
|
clangOmpss2-lto = lto.override { stdenv = final.stdenvClangOmpss2Nanos6; };
|
||||||
clangOmpss2-asan = asan.override { stdenv = final.stdenvClangOmpss2Nanos6; };
|
clangOmpss2-asan = asan.override { stdenv = final.stdenvClangOmpss2Nanos6; };
|
||||||
clangOmpss2-task = callPackage ./test/compilers/ompss2.nix {
|
clangOmpss2-task = callPackage ./test/compilers/ompss2.nix {
|
||||||
|
|||||||
@ -45,26 +45,35 @@ let
|
|||||||
installPhase = ''
|
installPhase = ''
|
||||||
awk -F': ' '\
|
awk -F': ' '\
|
||||||
BEGIN { print "[ {" } \
|
BEGIN { print "[ {" } \
|
||||||
NF==0 { empty=1; } \
|
NR>1 && /^Package: / { print "} {"; } \
|
||||||
NF && empty { print "} {"; empty=0; } \
|
|
||||||
/: / { printf "%s = \"%s\";\n", $1, $2 } \
|
/: / { printf "%s = \"%s\";\n", $1, $2 } \
|
||||||
END { print "} ]" }' $srcs > $out
|
END { print "} ]" }' $srcs > $out
|
||||||
'';
|
'';
|
||||||
};
|
};
|
||||||
|
|
||||||
findMatch = name:
|
aptPackages = import aptPackageIndex;
|
||||||
let
|
|
||||||
aptPackages = import aptPackageIndex;
|
|
||||||
matches = lib.filter (x: name == x.Package) aptPackages;
|
|
||||||
n = lib.length matches;
|
|
||||||
match = builtins.traceVerbose (name + " -- ${builtins.toString n}") (builtins.head matches);
|
|
||||||
|
|
||||||
apthost = "https://apt.repos.intel.com/oneapi/";
|
apthost = "https://apt.repos.intel.com/oneapi/";
|
||||||
|
|
||||||
|
getSum = pkgList: name:
|
||||||
|
let
|
||||||
|
matches = lib.filter (x: name == x.Package) pkgList;
|
||||||
|
#n = lib.length matches;
|
||||||
|
#match = builtins.trace (name + " -- ${builtins.toString n}") (lib.elemAt matches 0);
|
||||||
|
match = lib.elemAt matches 0;
|
||||||
in
|
in
|
||||||
{
|
match.SHA256;
|
||||||
url = apthost + match.Filename;
|
|
||||||
sha256 = match.SHA256;
|
getUrl = pkgList: name:
|
||||||
};
|
let
|
||||||
|
matches = lib.filter (x: name == x.Package) pkgList;
|
||||||
|
#match = assert lib.length matches == 1; lib.elemAt matches 0;
|
||||||
|
n = lib.length matches;
|
||||||
|
match =
|
||||||
|
#builtins.trace (name + " -- n=${builtins.toString n}")
|
||||||
|
(lib.elemAt matches 0);
|
||||||
|
in
|
||||||
|
apthost + match.Filename;
|
||||||
|
|
||||||
uncompressDebs = debs: name: stdenv.mkDerivation {
|
uncompressDebs = debs: name: stdenv.mkDerivation {
|
||||||
name = name;
|
name = name;
|
||||||
@ -82,7 +91,10 @@ let
|
|||||||
|
|
||||||
joinDebs = name: names:
|
joinDebs = name: names:
|
||||||
let
|
let
|
||||||
debs = builtins.map (x: builtins.fetchurl (findMatch x)) names;
|
urls = builtins.map (x: getUrl aptPackages x) names;
|
||||||
|
sums = builtins.map (x: getSum aptPackages x) names;
|
||||||
|
getsrc = url: sha256: builtins.fetchurl { inherit url sha256; };
|
||||||
|
debs = lib.zipListsWith getsrc urls sums;
|
||||||
in
|
in
|
||||||
uncompressDebs debs "${name}-source";
|
uncompressDebs debs "${name}-source";
|
||||||
|
|
||||||
@ -441,7 +453,7 @@ let
|
|||||||
'';
|
'';
|
||||||
};
|
};
|
||||||
|
|
||||||
ifort-wrapper = wrapIntel {
|
ifort-wrapper = wrapIntel rec {
|
||||||
cc = intel-compiler-fortran;
|
cc = intel-compiler-fortran;
|
||||||
mygcc = gcc;
|
mygcc = gcc;
|
||||||
extraBuild = ''
|
extraBuild = ''
|
||||||
|
|||||||
File diff suppressed because it is too large
Load Diff
File diff suppressed because it is too large
Load Diff
File diff suppressed because it is too large
Load Diff
@ -1,205 +0,0 @@
|
|||||||
{ lib
|
|
||||||
, stdenv
|
|
||||||
, callPackage
|
|
||||||
, dpkg
|
|
||||||
, fetchurl
|
|
||||||
, recurseIntoAttrs
|
|
||||||
|
|
||||||
, sqlite
|
|
||||||
, elfutils
|
|
||||||
}:
|
|
||||||
|
|
||||||
let
|
|
||||||
inherit (builtins)
|
|
||||||
elem attrNames attrValues concatMap filter fromJSON getAttr groupBy head isNull listToAttrs map mapAttrs readFile replaceStrings splitVersion ;
|
|
||||||
inherit (lib)
|
|
||||||
converge findFirst groupBy' hasPrefix optional pipe take toInt toList versionAtLeast versionOlder ;
|
|
||||||
|
|
||||||
aptData = fromJSON (readFile ./packages.json);
|
|
||||||
|
|
||||||
# Compare versions in debian control file syntax
|
|
||||||
# See: https://www.debian.org/doc/debian-policy/ch-relationships.html#syntax-of-relationship-fields
|
|
||||||
#
|
|
||||||
# NOTE: this is not a proper version comparison
|
|
||||||
#
|
|
||||||
# A proper version solver, should aggregate dependencies with the same name
|
|
||||||
# and compute the constraint (e.g. a (>= 2) a (<< 5) -> 2 <= a << 5)
|
|
||||||
#
|
|
||||||
# But in the intel repo, there are no such "duplicated" dependencies to specify
|
|
||||||
# upper limits, which leads to issues when intel-hpckit-2021 depends on things
|
|
||||||
# like intel-basekit >= 2021.1.0-2403 and we end up installing the newest
|
|
||||||
# basekit instead of the one from 2021.
|
|
||||||
#
|
|
||||||
# To mitigate this, >= is set to take the latest version with matching major
|
|
||||||
# and minor (only revision and patch are allowed to change)
|
|
||||||
compareVersions = got: kind: want:
|
|
||||||
let
|
|
||||||
g0 = take 2 (splitVersion got);
|
|
||||||
w0 = take 2 (splitVersion want);
|
|
||||||
in if isNull want then true
|
|
||||||
else if kind == "=" then got == want
|
|
||||||
else if kind == "<<" then versionOlder got want
|
|
||||||
else if kind == "<=" then versionAtLeast want got
|
|
||||||
else if kind == ">>" then versionOlder want got
|
|
||||||
else if kind == ">=" then (g0 == w0) && versionAtLeast got want # always match major version
|
|
||||||
else throw "unknown operation: ${kind}";
|
|
||||||
|
|
||||||
findMatching = { pname, kind, version }:
|
|
||||||
findFirst (x: pname == x.pname && compareVersions x.version kind version) null aptData;
|
|
||||||
|
|
||||||
isIntel = pkg: (hasPrefix "intel-" pkg.pname);
|
|
||||||
|
|
||||||
expandDeps = pkg: (map findMatching (filter isIntel pkg.dependencies)) ++ (optional (pkg.size != 0) pkg);
|
|
||||||
|
|
||||||
# get the oldest by major version. If they have the same major version, take
|
|
||||||
# the newest. This prevents most issues with resolutions
|
|
||||||
# versionOlder b a -> true if b is older than a (b `older` a)
|
|
||||||
getNewerInMajor = a: b: let
|
|
||||||
va = a.version;
|
|
||||||
vb = b.version;
|
|
||||||
va0 = head (splitVersion va);
|
|
||||||
vb0 = head (splitVersion vb);
|
|
||||||
in
|
|
||||||
if isNull a then b
|
|
||||||
else if va0 != vb0 then
|
|
||||||
if va0 > vb0 then b else a
|
|
||||||
else if versionOlder vb va then a else b;
|
|
||||||
removeDups = l: attrValues (groupBy' getNewerInMajor null (getAttr "provides") l);
|
|
||||||
|
|
||||||
_resolveDeps = converge (l: removeDups (concatMap expandDeps l));
|
|
||||||
resolveDeps = pkg: let deps = _resolveDeps (toList pkg);
|
|
||||||
namedDeps = (map (x: "${x.pname}-${x.version}") deps);
|
|
||||||
in builtins.traceVerbose (builtins.deepSeq namedDeps namedDeps) deps;
|
|
||||||
|
|
||||||
blacklist = [
|
|
||||||
"intel-basekit-env"
|
|
||||||
"intel-basekit-getting-started"
|
|
||||||
"intel-hpckit-env"
|
|
||||||
"intel-hpckit-getting-started"
|
|
||||||
"intel-oneapi-advisor"
|
|
||||||
"intel-oneapi-common-licensing"
|
|
||||||
"intel-oneapi-common-oneapi-vars"
|
|
||||||
"intel-oneapi-common-vars"
|
|
||||||
"intel-oneapi-compiler-cpp-eclipse-cfg"
|
|
||||||
"intel-oneapi-compiler-dpcpp-eclipse-cfg"
|
|
||||||
"intel-oneapi-condaindex"
|
|
||||||
"intel-oneapi-dev-utilities-eclipse-cfg"
|
|
||||||
"intel-oneapi-dpcpp-ct-eclipse-cfg"
|
|
||||||
"intel-oneapi-eclipse-ide"
|
|
||||||
"intel-oneapi-hpc-toolkit-getting-started"
|
|
||||||
"intel-oneapi-icc-eclipse-plugin-cpp"
|
|
||||||
"intel-oneapi-vtune"
|
|
||||||
"intel-oneapi-vtune-eclipse-plugin-vtune"
|
|
||||||
];
|
|
||||||
|
|
||||||
isInBlacklist = pkg: elem pkg.provides blacklist;
|
|
||||||
removeBlacklist = filter (e: !(isInBlacklist e));
|
|
||||||
|
|
||||||
dpkgExtractAll = pname: version: {srcs, deps}: stdenv.mkDerivation {
|
|
||||||
inherit pname version srcs;
|
|
||||||
|
|
||||||
nativeBuildInputs = [ dpkg ];
|
|
||||||
phases = [ "installPhase" ];
|
|
||||||
|
|
||||||
passthru = { inherit deps; };
|
|
||||||
|
|
||||||
installPhase = ''
|
|
||||||
mkdir -p $out
|
|
||||||
for src in $srcs; do
|
|
||||||
echo "Unpacking $src"
|
|
||||||
dpkg -x $src $out
|
|
||||||
done
|
|
||||||
'';
|
|
||||||
};
|
|
||||||
|
|
||||||
apthost = "https://apt.repos.intel.com/oneapi/";
|
|
||||||
fetchDeb = p: fetchurl { url = apthost + p.filename; inherit (p) sha256; };
|
|
||||||
|
|
||||||
buildIntel = pkg: pipe pkg [
|
|
||||||
resolveDeps
|
|
||||||
removeBlacklist
|
|
||||||
(l: {srcs = map fetchDeb l; deps = l; })
|
|
||||||
(dpkgExtractAll "${pkg.provides}-extracted" pkg.version)
|
|
||||||
];
|
|
||||||
|
|
||||||
findHpcKit = year: findMatching { pname = "intel-hpckit"; kind = "<<"; version = toString (year+1); };
|
|
||||||
years = map toInt (attrNames components);
|
|
||||||
|
|
||||||
patchIntel = callPackage ./patch_intel.nix { };
|
|
||||||
|
|
||||||
# Version information for each hpckit. This is used to normalize the paths
|
|
||||||
# so that files are in $out/{bin,lib,include...} instead of all over the place
|
|
||||||
# in $out/opt/intel/oneapi/*/*/{...}.
|
|
||||||
#
|
|
||||||
# The most important is the compiler component, which is used to build the
|
|
||||||
# stdenv for the hpckit.
|
|
||||||
#
|
|
||||||
# NOTE: this have to be manually specified, so we can avoid IFD. To add a
|
|
||||||
# new version, add a new field with an empty attrset, (e.g. "2026" = {}; ),
|
|
||||||
# build hpckit_2026.unpatched and use the values from
|
|
||||||
# result/opt/intel/oneapi/* to populate the attrset.
|
|
||||||
#
|
|
||||||
# WARN: if there are more than one version in the folders of the unpatched
|
|
||||||
# components, our dependency resolution hacks have probably failed and the
|
|
||||||
# package set may be broken.
|
|
||||||
components = {
|
|
||||||
"2025" = {
|
|
||||||
ishmem = "1.3";
|
|
||||||
pti = "0.12";
|
|
||||||
tcm = "1.3";
|
|
||||||
umf = "0.10";
|
|
||||||
|
|
||||||
ccl = "2021.15";
|
|
||||||
compiler = "2025.1";
|
|
||||||
dal = "2025.5";
|
|
||||||
debugger = "2025.1";
|
|
||||||
dev-utilities = "2025.1";
|
|
||||||
dnnl = "2025.1";
|
|
||||||
dpcpp-ct = "2025.1";
|
|
||||||
dpl = "2022.8";
|
|
||||||
ipp = "2022.1";
|
|
||||||
ippcp = "2025.1";
|
|
||||||
mkl = "2025.1";
|
|
||||||
mpi = "2021.15";
|
|
||||||
tbb = "2022.1";
|
|
||||||
};
|
|
||||||
"2024" = {
|
|
||||||
tcm = "1.1";
|
|
||||||
|
|
||||||
ccl = "2021.13";
|
|
||||||
compiler = "2024.2";
|
|
||||||
dal = "2024.6";
|
|
||||||
debugger = "2024.2";
|
|
||||||
dev-utilities = "2024.2";
|
|
||||||
diagnostics = "2024.2";
|
|
||||||
dnnl = "2024.2";
|
|
||||||
dpcpp-ct = "2024.2";
|
|
||||||
dpl = "2022.6";
|
|
||||||
ipp = "2021.12";
|
|
||||||
ippcp = "2021.12";
|
|
||||||
mkl = "2024.2";
|
|
||||||
mpi = "2021.13";
|
|
||||||
tbb = "2021.13";
|
|
||||||
|
|
||||||
extraPackages = [
|
|
||||||
sqlite
|
|
||||||
elfutils
|
|
||||||
];
|
|
||||||
};
|
|
||||||
};
|
|
||||||
|
|
||||||
replaceDots = replaceStrings ["."] ["_"];
|
|
||||||
|
|
||||||
in recurseIntoAttrs (listToAttrs (map (year: let
|
|
||||||
year_str = toString year;
|
|
||||||
in {
|
|
||||||
name = "hpckit_${year_str}";
|
|
||||||
value = patchIntel {unpatched = buildIntel (findHpcKit year); components = components.${year_str}; };
|
|
||||||
}) years)) // {
|
|
||||||
apt = pipe aptData [
|
|
||||||
(groupBy (p: replaceDots p.provides))
|
|
||||||
(mapAttrs (_: l: listToAttrs (map (pkg: { name = replaceDots ("v" + pkg.version); value = pkg; }) l)))
|
|
||||||
] ;
|
|
||||||
|
|
||||||
inherit resolveDeps patchIntel buildIntel;
|
|
||||||
}
|
|
||||||
@ -1,177 +0,0 @@
|
|||||||
{ stdenv
|
|
||||||
, stdenvNoCC
|
|
||||||
, lib
|
|
||||||
, symlinkJoin
|
|
||||||
, autoPatchelfHook
|
|
||||||
, wrapCCWith
|
|
||||||
, overrideCC
|
|
||||||
, gcc13
|
|
||||||
, hwloc
|
|
||||||
, libelf
|
|
||||||
, libffi_3_3
|
|
||||||
, libpsm2
|
|
||||||
, libuuid
|
|
||||||
, libxml2
|
|
||||||
, numactl
|
|
||||||
, ocl-icd
|
|
||||||
, openssl
|
|
||||||
, python3
|
|
||||||
, rdma-core
|
|
||||||
, ucx
|
|
||||||
, zlib
|
|
||||||
, makeOverridable
|
|
||||||
, recurseIntoAttrs
|
|
||||||
}:
|
|
||||||
|
|
||||||
makeOverridable (
|
|
||||||
{
|
|
||||||
unpatched,
|
|
||||||
components ? { },
|
|
||||||
extraPackages ? components.extraPackages or []
|
|
||||||
}:
|
|
||||||
|
|
||||||
let
|
|
||||||
inherit (builtins) attrValues filter mapAttrs removeAttrs;
|
|
||||||
|
|
||||||
gcc = gcc13;
|
|
||||||
|
|
||||||
__components = removeAttrs components ["extraPackages"];
|
|
||||||
_components = __components;
|
|
||||||
# _components = lib.traceSeqN 2 {
|
|
||||||
# inherit unpatched __components;
|
|
||||||
# deps = builtins.map (x: "${x.pname}-${x.version}") unpatched.deps;
|
|
||||||
# } __components;
|
|
||||||
|
|
||||||
wrapIntel = { cc, extraBuildCommands ? "", extraInstall ? "", ... }@args:
|
|
||||||
let
|
|
||||||
targetConfig = stdenv.targetPlatform.config;
|
|
||||||
in (wrapCCWith {
|
|
||||||
inherit cc;
|
|
||||||
extraBuildCommands = ''
|
|
||||||
echo "-L${gcc.cc}/lib/gcc/${targetConfig}/${gcc.version}" >> $out/nix-support/cc-ldflags
|
|
||||||
echo "-L${gcc.cc.lib}/lib" >> $out/nix-support/cc-ldflags
|
|
||||||
echo "-L${cc}/lib" >> $out/nix-support/cc-ldflags
|
|
||||||
|
|
||||||
# echo "--gcc-toolchain=${gcc.cc}" >> $out/nix-support/libcxx-cxxflags
|
|
||||||
echo "--gcc-toolchain=${gcc.cc}" >> $out/nix-support/cc-cflags
|
|
||||||
|
|
||||||
# For some reason, If we don't resolve the realpath things go wrong
|
|
||||||
for stddef in ${cc}/lib/clang/*/include/stddef.h ; do
|
|
||||||
dir=$(dirname $(realpath "$stddef"))
|
|
||||||
echo "-isystem $dir" >> $out/nix-support/cc-cflags
|
|
||||||
done
|
|
||||||
|
|
||||||
echo "-isystem ${cc}/include" >> $out/nix-support/cc-cflags
|
|
||||||
echo "-isystem ${cc}/include/intel64" >> $out/nix-support/cc-cflags
|
|
||||||
|
|
||||||
for dir in ${gcc.cc}/lib/gcc/${targetConfig}/*/include; do
|
|
||||||
echo "-isystem $dir" >> $out/nix-support/cc-cflags
|
|
||||||
done
|
|
||||||
|
|
||||||
for dir in ${gcc.cc}/include/c++/*; do
|
|
||||||
echo "-isystem $dir" >> $out/nix-support/libcxx-cxxflags
|
|
||||||
done
|
|
||||||
for dir in ${gcc.cc}/include/c++/*/${targetConfig}; do
|
|
||||||
echo "-isystem $dir" >> $out/nix-support/libcxx-cxxflags
|
|
||||||
done
|
|
||||||
|
|
||||||
# FIXME: We should find a better way to modify the PATH instead of using
|
|
||||||
# this ugly hack. See https://jungle.bsc.es/git/rarias/bscpkgs/issues/9
|
|
||||||
echo 'path_backup="${gcc.cc}/bin:$path_backup"' >> $out/nix-support/cc-wrapper-hook
|
|
||||||
|
|
||||||
# Disable hardening by default
|
|
||||||
echo "" > $out/nix-support/add-hardening.sh
|
|
||||||
'' + extraBuildCommands;
|
|
||||||
} // (removeAttrs args ["cc" "extraBuildCommands" "extraInstall"])
|
|
||||||
).overrideAttrs (old: {
|
|
||||||
installPhase = old.installPhase + extraInstall;
|
|
||||||
});
|
|
||||||
|
|
||||||
in
|
|
||||||
stdenvNoCC.mkDerivation (finalAttrs: {
|
|
||||||
pname = lib.removeSuffix "-extracted" unpatched.pname;
|
|
||||||
inherit (unpatched) version;
|
|
||||||
src = unpatched;
|
|
||||||
|
|
||||||
phases = [ "installPhase" "fixupPhase" ];
|
|
||||||
buildInputs = [
|
|
||||||
libffi_3_3
|
|
||||||
libelf
|
|
||||||
libxml2
|
|
||||||
hwloc
|
|
||||||
numactl
|
|
||||||
libuuid
|
|
||||||
libpsm2
|
|
||||||
zlib
|
|
||||||
ocl-icd
|
|
||||||
rdma-core
|
|
||||||
ucx
|
|
||||||
openssl
|
|
||||||
python3
|
|
||||||
stdenv.cc.cc.lib
|
|
||||||
] ++ extraPackages;
|
|
||||||
|
|
||||||
autoPatchelfIgnoreMissingDeps = [ "libhwloc.so.5" "libcuda.so.1" "libze_loader.so.1" ];
|
|
||||||
|
|
||||||
dontCheckForBrokenSymlinks = true;
|
|
||||||
|
|
||||||
nativeBuildInputs = [ autoPatchelfHook ];
|
|
||||||
installPhase = ''
|
|
||||||
cp -r $src/opt/intel/oneapi/ $out
|
|
||||||
'';
|
|
||||||
|
|
||||||
passthru = let
|
|
||||||
pkgs = mapAttrs
|
|
||||||
(folder: version: symlinkJoin {
|
|
||||||
pname = "intel-${folder}";
|
|
||||||
inherit version;
|
|
||||||
paths = ["${finalAttrs.finalPackage}/${folder}/${version}"];
|
|
||||||
})
|
|
||||||
_components;
|
|
||||||
in pkgs // {
|
|
||||||
inherit unpatched;
|
|
||||||
pkgs = recurseIntoAttrs pkgs;
|
|
||||||
components = _components;
|
|
||||||
|
|
||||||
# This contains all packages properly symlinked into toplevel directories
|
|
||||||
# in $out.
|
|
||||||
#
|
|
||||||
# NOTE: there are clashes with packages that have symlinks outside their
|
|
||||||
# scope (libtcm and env/vars.sh)
|
|
||||||
all = symlinkJoin {
|
|
||||||
pname = finalAttrs.finalPackage + "-symlinked";
|
|
||||||
inherit (finalAttrs.finalPackage) version;
|
|
||||||
paths = filter lib.isDerivation (attrValues finalAttrs.finalPackage.pkgs);
|
|
||||||
};
|
|
||||||
|
|
||||||
stdenv = overrideCC stdenv finalAttrs.finalPackage.cc;
|
|
||||||
|
|
||||||
cc = wrapIntel {
|
|
||||||
cc = finalAttrs.finalPackage.pkgs.compiler;
|
|
||||||
extraBuildCommands = ''
|
|
||||||
wrap icx $wrapper $ccPath/icx
|
|
||||||
wrap icpx $wrapper $ccPath/icpx
|
|
||||||
wrap ifx $wrapper $ccPath/ifx
|
|
||||||
|
|
||||||
ln -s $out/bin/icpx $out/bin/c++
|
|
||||||
ln -s $out/bin/icx $out/bin/cc
|
|
||||||
|
|
||||||
# Use this to detect when a compiler subprocess is called
|
|
||||||
# from icpx (--fsycl-host-compiler)
|
|
||||||
echo 'export "NIX_CC_WRAPPER_INTEL=1"' >> $out/nix-support/cc-wrapper-hook
|
|
||||||
|
|
||||||
sed -i 's/.*isCxx=0/isCxx=1/' $out/bin/icpx
|
|
||||||
|
|
||||||
# oneMath looks for sycl libraries in bin/../lib
|
|
||||||
ln -s ${finalAttrs.finalPackage.pkgs.compiler}/lib $out/lib
|
|
||||||
ln -s ${finalAttrs.finalPackage.pkgs.compiler}/include $out/include
|
|
||||||
'';
|
|
||||||
extraInstall = ''
|
|
||||||
export named_cc="icx"
|
|
||||||
export named_cxx="icpx"
|
|
||||||
export named_fc="ifx"
|
|
||||||
'';
|
|
||||||
};
|
|
||||||
};
|
|
||||||
|
|
||||||
}))
|
|
||||||
@ -1,29 +0,0 @@
|
|||||||
#!/usr/bin/env -S jq -f
|
|
||||||
|
|
||||||
def extract_fields: {
|
|
||||||
pname : .Package,
|
|
||||||
version : .Version,
|
|
||||||
provides : .Package | sub("[0-9.-]*$"; ""),
|
|
||||||
filename : .Filename,
|
|
||||||
size : ."Installed-Size" | tonumber,
|
|
||||||
sha256 : .SHA256,
|
|
||||||
dependencies : .Depends,
|
|
||||||
} ;
|
|
||||||
|
|
||||||
# parses dependencies into a list of [{.pname, .kind, .version}]
|
|
||||||
# some dependencies do not have a version specified, in which case, version = version = null
|
|
||||||
#
|
|
||||||
# example dependencies:
|
|
||||||
# intel-oneapi-common-vars (>= 2023.0.0-25325), intel-oneapi-common-licensing-2023.0.0
|
|
||||||
|
|
||||||
def split_dependencies : map(try(.dependencies |= split(",\\s?"; "")) // .dependencies |= []) ;
|
|
||||||
|
|
||||||
def match_version : capture("(?<pname>[a-zA-Z0-9_\\-.]*) *(\\((?<kind>[<>=]*) *(?<version>.*)\\))?"; "") ;
|
|
||||||
|
|
||||||
def parse_dependencies : map_values(.dependencies.[] |= match_version) ;
|
|
||||||
|
|
||||||
def sort_version_decreasing : sort_by(.version | split("[-.]"; "") | map(tonumber)) | reverse ;
|
|
||||||
|
|
||||||
map(extract_fields) | split_dependencies | parse_dependencies | sort_version_decreasing
|
|
||||||
|
|
||||||
# [.[] | select(.pname == "intel-hpckit") | .version]
|
|
||||||
@ -1,29 +0,0 @@
|
|||||||
#!/usr/bin/env -S awk -f
|
|
||||||
|
|
||||||
BEGIN {
|
|
||||||
FS=": "
|
|
||||||
|
|
||||||
prev_empty=1
|
|
||||||
t=" "
|
|
||||||
|
|
||||||
print "[ {"
|
|
||||||
}
|
|
||||||
|
|
||||||
!NF { # empty line, update separator so next non empty line closes the dict
|
|
||||||
prev_empty=1
|
|
||||||
t="},\n{ "
|
|
||||||
next # skip line (we won't match anything else)
|
|
||||||
}
|
|
||||||
|
|
||||||
{
|
|
||||||
printf t "\"%s\" : \"%s\"\n", $1, $2
|
|
||||||
|
|
||||||
if (prev_empty) {
|
|
||||||
# we were the first after a group of empty lines, following ones have to
|
|
||||||
# have a comma
|
|
||||||
prev_empty=0
|
|
||||||
t=", "
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
END { print "} ]" }
|
|
||||||
@ -2,5 +2,3 @@
|
|||||||
|
|
||||||
curl https://apt.repos.intel.com/oneapi/dists/all/main/binary-amd64/Packages -o amd64-packages
|
curl https://apt.repos.intel.com/oneapi/dists/all/main/binary-amd64/Packages -o amd64-packages
|
||||||
curl https://apt.repos.intel.com/oneapi/dists/all/main/binary-all/Packages -o all-packages
|
curl https://apt.repos.intel.com/oneapi/dists/all/main/binary-all/Packages -o all-packages
|
||||||
|
|
||||||
cat amd64-packages all-packages | ./toJson.awk | ./process.jq >packages.json
|
|
||||||
|
|||||||
@ -2,9 +2,7 @@
|
|||||||
stdenv
|
stdenv
|
||||||
, lib
|
, lib
|
||||||
, gcc
|
, gcc
|
||||||
, gcc13
|
|
||||||
, clangOmpss2Unwrapped
|
, clangOmpss2Unwrapped
|
||||||
, writeShellScript
|
|
||||||
, openmp ? null
|
, openmp ? null
|
||||||
, wrapCCWith
|
, wrapCCWith
|
||||||
, llvmPackages_latest
|
, llvmPackages_latest
|
||||||
@ -39,7 +37,10 @@ let
|
|||||||
inherit gcc;
|
inherit gcc;
|
||||||
cc = clangOmpss2Unwrapped;
|
cc = clangOmpss2Unwrapped;
|
||||||
gccVersion = with versions; let v = gcc.version; in concatStringsSep "." [(major v) (minor v) (patch v)];
|
gccVersion = with versions; let v = gcc.version; in concatStringsSep "." [(major v) (minor v) (patch v)];
|
||||||
|
in wrapCCWith {
|
||||||
|
inherit cc bintools;
|
||||||
|
# extraPackages adds packages to depsTargetTargetPropagated
|
||||||
|
extraPackages = optional (openmp != null) openmp;
|
||||||
extraBuildCommands = ''
|
extraBuildCommands = ''
|
||||||
echo "-target ${targetConfig}" >> $out/nix-support/cc-cflags
|
echo "-target ${targetConfig}" >> $out/nix-support/cc-cflags
|
||||||
echo "-B${gcc.cc}/lib/gcc/${targetConfig}/${gccVersion}" >> $out/nix-support/cc-cflags
|
echo "-B${gcc.cc}/lib/gcc/${targetConfig}/${gccVersion}" >> $out/nix-support/cc-cflags
|
||||||
@ -56,51 +57,14 @@ let
|
|||||||
echo "--gcc-toolchain=${gcc}" >> $out/nix-support/cc-cflags
|
echo "--gcc-toolchain=${gcc}" >> $out/nix-support/cc-cflags
|
||||||
|
|
||||||
wrap clang++ $wrapper $ccPath/clang++
|
wrap clang++ $wrapper $ccPath/clang++
|
||||||
|
|
||||||
|
'' + optionalString (openmp != null) ''
|
||||||
|
echo "export OPENMP_RUNTIME=${ompname}" >> $out/nix-support/cc-wrapper-hook
|
||||||
|
'' + optionalString (ompss2rt != null) ''
|
||||||
|
echo "export OMPSS2_RUNTIME=${rtname}" >> $out/nix-support/cc-wrapper-hook
|
||||||
|
echo "export ${homevar}=${ompss2rt}" >> $out/nix-support/cc-wrapper-hook
|
||||||
|
'' + optionalString (ompss2rt != null && ompss2rt.pname == "nodes") ''
|
||||||
|
echo "export NOSV_HOME=${ompss2rt.nosv}" >> $out/nix-support/cc-wrapper-hook
|
||||||
'';
|
'';
|
||||||
|
}
|
||||||
|
|
||||||
envExports = lib.optionalString (openmp != null) ''
|
|
||||||
echo "export OPENMP_RUNTIME=${ompname}" >> $out/nix-support/cc-wrapper-hook
|
|
||||||
'' + optionalString (ompss2rt != null) ''
|
|
||||||
echo "export OMPSS2_RUNTIME=${rtname}" >> $out/nix-support/cc-wrapper-hook
|
|
||||||
echo "export ${homevar}=${ompss2rt}" >> $out/nix-support/cc-wrapper-hook
|
|
||||||
'' + optionalString (ompss2rt != null && ompss2rt.pname == "nodes") ''
|
|
||||||
echo "export NOSV_HOME=${ompss2rt.nosv}" >> $out/nix-support/cc-wrapper-hook
|
|
||||||
'';
|
|
||||||
|
|
||||||
extraPackages = optional (openmp != null) openmp;
|
|
||||||
|
|
||||||
wrappedCC = wrapCCWith {
|
|
||||||
# extraPackages adds packages to depsTargetTargetPropagated
|
|
||||||
inherit cc bintools extraPackages;
|
|
||||||
extraBuildCommands = extraBuildCommands + envExports;
|
|
||||||
};
|
|
||||||
|
|
||||||
resetIntelCCFlags = let tconf = builtins.replaceStrings ["-"] ["_"] targetConfig;
|
|
||||||
in writeShellScript "remove-intel.sh" ''
|
|
||||||
if [ "''${NIX_CC_WRAPPER_INTEL:-0}" = 1 ]; then
|
|
||||||
unset NIX_CFLAGS_COMPILE_${tconf}
|
|
||||||
unset NIX_CC_WRAPPER_FLAGS_SET_${tconf}
|
|
||||||
|
|
||||||
if (( "''${NIX_DEBUG:-0}" >= 1 )); then
|
|
||||||
echo "ompss2: cleaned NIX_CFLAGS_COMPILE_${tconf} (invokation from intel compiler detected)"
|
|
||||||
fi
|
|
||||||
fi
|
|
||||||
'';
|
|
||||||
|
|
||||||
intelExtraBuildCommands = ''
|
|
||||||
sed -i 's|# Flirting.*|source ${resetIntelCCFlags}\n\n&|' $out/bin/clang
|
|
||||||
sed -i 's|# Flirting.*|source ${resetIntelCCFlags}\n\n&|' $out/bin/clang++
|
|
||||||
'';
|
|
||||||
|
|
||||||
wrappedCCIntel = wrapCCWith {
|
|
||||||
inherit bintools extraPackages;
|
|
||||||
cc = cc.override { gcc = gcc13; }; # Intel uses gcc13, so we have to match it
|
|
||||||
# extraPackages adds packages to depsTargetTargetPropagated
|
|
||||||
extraBuildCommands = intelExtraBuildCommands + envExports;
|
|
||||||
};
|
|
||||||
|
|
||||||
in wrappedCC.overrideAttrs (oldAttrs: {
|
|
||||||
passthru = oldAttrs.passthru // {
|
|
||||||
forIcpx = wrappedCCIntel;
|
|
||||||
};
|
|
||||||
})
|
|
||||||
|
|||||||
@ -1,84 +0,0 @@
|
|||||||
{ lib
|
|
||||||
, fetchFromGitHub
|
|
||||||
, cmake
|
|
||||||
, withCFlags
|
|
||||||
, intelPackages
|
|
||||||
|
|
||||||
, mklSupport ? true
|
|
||||||
|
|
||||||
, config
|
|
||||||
|
|
||||||
, cudaSupport ? config.cudaSupport
|
|
||||||
, cudaPackages ? { }
|
|
||||||
|
|
||||||
, rocmSupport ? config.rocmSupport
|
|
||||||
, hipTargets ? null # only one target at a time supported
|
|
||||||
, rocmPackages ? { }
|
|
||||||
}:
|
|
||||||
|
|
||||||
let
|
|
||||||
# rocmSupport is not enough, we need a specific target
|
|
||||||
enableHip = rocmSupport && hipTargets != null;
|
|
||||||
|
|
||||||
stdenv = withCFlags (lib.optionals cudaSupport [ "--cuda-path=${cudaPackages.cudatoolkit}" ]) intelPackages.stdenv;
|
|
||||||
in
|
|
||||||
|
|
||||||
# at least one backend has to be enabled
|
|
||||||
assert mklSupport || cudaSupport || enableHip;
|
|
||||||
|
|
||||||
stdenv.mkDerivation rec {
|
|
||||||
pname = "oneMath";
|
|
||||||
version = "0.8";
|
|
||||||
|
|
||||||
src = fetchFromGitHub {
|
|
||||||
owner = "uxlfoundation";
|
|
||||||
repo = "oneMath";
|
|
||||||
rev = "v${version}";
|
|
||||||
sha256 = "sha256-xK8lKI3oqKlx3xtvdScpMq+HXAuoYCP0BZdkEqnJP5o=";
|
|
||||||
};
|
|
||||||
|
|
||||||
cmakeFlags = [
|
|
||||||
(lib.cmakeBool "ENABLE_MKLCPU_BACKEND" mklSupport)
|
|
||||||
(lib.cmakeBool "ENABLE_MKLGPU_BACKEND" mklSupport)
|
|
||||||
|
|
||||||
(lib.cmakeBool "ENABLE_CUBLAS_BACKEND" cudaSupport)
|
|
||||||
(lib.cmakeBool "ENABLE_CUFFT_BACKEND" cudaSupport)
|
|
||||||
(lib.cmakeBool "ENABLE_CURAND_BACKEND" cudaSupport)
|
|
||||||
(lib.cmakeBool "ENABLE_CUSOLVER_BACKEND" cudaSupport)
|
|
||||||
(lib.cmakeBool "ENABLE_CUSPARSE_BACKEND" cudaSupport)
|
|
||||||
|
|
||||||
(lib.cmakeBool "ENABLE_ROCBLAS_BACKEND" enableHip)
|
|
||||||
(lib.cmakeBool "ENABLE_ROCFFT_BACKEND" enableHip)
|
|
||||||
(lib.cmakeBool "ENABLE_ROCSOLVER_BACKEND" enableHip)
|
|
||||||
(lib.cmakeBool "ENABLE_ROCRAND_BACKEND" enableHip)
|
|
||||||
(lib.cmakeBool "ENABLE_ROCSPARSE_BACKEND" enableHip)
|
|
||||||
|
|
||||||
(lib.cmakeBool "BUILD_FUNCTIONAL_TESTS" false)
|
|
||||||
(lib.cmakeBool "BUILD_EXAMPLES" false)
|
|
||||||
] ++ lib.optionals enableHip [
|
|
||||||
(lib.cmakeFeature "HIP_TARGETS" hipTargets)
|
|
||||||
];
|
|
||||||
|
|
||||||
nativeBuildInputs = [ cmake ];
|
|
||||||
buildInputs = lib.optionals (mklSupport) [
|
|
||||||
intelPackages.mkl
|
|
||||||
intelPackages.tbb
|
|
||||||
] ++ lib.optionals (enableHip) [
|
|
||||||
rocmPackages.rocmPath
|
|
||||||
rocmPackages.rocblas
|
|
||||||
rocmPackages.rocfft
|
|
||||||
rocmPackages.rocsolver
|
|
||||||
rocmPackages.rocrand
|
|
||||||
rocmPackages.rocsparse
|
|
||||||
] ++ lib.optionals (cudaSupport) [
|
|
||||||
(lib.getDev cudaPackages.cuda_cudart)
|
|
||||||
cudaPackages.cudatoolkit
|
|
||||||
|
|
||||||
cudaPackages.libcublas
|
|
||||||
cudaPackages.libcurand
|
|
||||||
cudaPackages.libcufft
|
|
||||||
cudaPackages.libcusparse
|
|
||||||
cudaPackages.libcusolver
|
|
||||||
];
|
|
||||||
|
|
||||||
}
|
|
||||||
@ -1,4 +1,4 @@
|
|||||||
{ writeText, intelPackages, nodes, nosv, clangOmpss2Nodes, strace }:
|
{ writeText, intelPackages, nodes, nosv, clangOmpss2Nodes, clangOmpss2Unwrapped, clangOmpss2, strace }:
|
||||||
|
|
||||||
let
|
let
|
||||||
hello_cpp = writeText "hello.cpp" ''
|
hello_cpp = writeText "hello.cpp" ''
|
||||||
@ -44,10 +44,6 @@ intelPackages.stdenv.mkDerivation {
|
|||||||
dontUnpack = true;
|
dontUnpack = true;
|
||||||
dontConfigure = true;
|
dontConfigure = true;
|
||||||
|
|
||||||
# NODES requires access to /sys/devices to request NUMA information. It will
|
|
||||||
# fail to run otherwise, so we disable the sandbox for this test.
|
|
||||||
__noChroot = true;
|
|
||||||
|
|
||||||
env.NODES_HOME = nodes;
|
env.NODES_HOME = nodes;
|
||||||
|
|
||||||
NIX_DEBUG = 0;
|
NIX_DEBUG = 0;
|
||||||
@ -59,7 +55,7 @@ intelPackages.stdenv.mkDerivation {
|
|||||||
command -v $CXX
|
command -v $CXX
|
||||||
|
|
||||||
icpx -Wno-deprecated-declarations -fsycl \
|
icpx -Wno-deprecated-declarations -fsycl \
|
||||||
-fsycl-host-compiler=${clangOmpss2Nodes.forIcpx}/bin/clang++ \
|
-fsycl-host-compiler=${clangOmpss2Nodes}/bin/clang++ \
|
||||||
-fsycl-host-compiler-options='-Wno-deprecated-declarations -fompss-2=libnodes' \
|
-fsycl-host-compiler-options='-Wno-deprecated-declarations -fompss-2=libnodes' \
|
||||||
-lnodes -lnosv \
|
-lnodes -lnosv \
|
||||||
$NODES_HOME/lib/nodes-main-wrapper.o \
|
$NODES_HOME/lib/nodes-main-wrapper.o \
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user