Add buildArch flag to translateLocally for controlling AVX512 (#385549)
This commit is contained in:
@@ -12,6 +12,7 @@
|
||||
runCommand,
|
||||
translatelocally,
|
||||
translatelocally-models,
|
||||
buildArch ? "x86-64",
|
||||
}:
|
||||
|
||||
let
|
||||
@@ -65,6 +66,10 @@ stdenv.mkDerivation (finalAttrs: {
|
||||
cmakeFlags = [
|
||||
"-DBLAS_LIBRARIES=-lblas"
|
||||
"-DCBLAS_LIBRARIES=-lcblas"
|
||||
|
||||
# See the following for context:
|
||||
# https://github.com/NixOS/nixpkgs/pull/385549
|
||||
(lib.optionalString stdenv.hostPlatform.isx86_64 "-DBUILD_ARCH=${buildArch}")
|
||||
];
|
||||
|
||||
passthru.tests = {
|
||||
|
||||
Reference in New Issue
Block a user