Skip to content

Commit e64f871

Browse files
committed
nix-builder: remove the MKL override
This override stems from when we were still building Torch from source. When we did so, we had to ensure that the build aligned as closely as possible to upstream, meaning that we had to build it with MKL. Since we repackage binary wheels, this is not necessary anymore, because the MKL bits are baked into the wheel. With this change, we can hit the NixOS cache for more dependencies on x86_64-linux.
1 parent 0b3c8f4 commit e64f871

1 file changed

Lines changed: 0 additions & 8 deletions

File tree

nix-builder/overlay.nix

Lines changed: 0 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,5 @@
11
final: prev:
2-
let
3-
# For XPU we use MKL from the joined oneAPI toolkit.
4-
useMKL = final.stdenv.isx86_64 && !(final.config.xpuSupport or false);
5-
in
62
{
7-
# Use MKL for BLAS/LAPACK on x86_64.
8-
blas = if useMKL then prev.blas.override { blasProvider = prev.mkl; } else prev.blas;
9-
lapack = if useMKL then prev.lapack.override { lapackProvider = prev.mkl; } else prev.lapack;
10-
113
kernel-builder = final.callPackage ./pkgs/kernel-builder { };
124

135
cmakeNvccThreadsHook = final.callPackage ./pkgs/cmake-nvcc-threads-hook { };

0 commit comments

Comments
 (0)