From dedfb4af4e4ab6af7d3b762ca49588e7fc2989b3 Mon Sep 17 00:00:00 2001 From: David Holtz Date: Wed, 2 Sep 2026 14:59:34 -0400 Subject: [PATCH] fix: format nix arch files --- nix-builder/lib/extension/torch/arch.nix | 21 ++++++++++----------- nix-builder/lib/extension/tvm-ffi/arch.nix | 21 ++++++++++----------- 2 files changed, 20 insertions(+), 22 deletions(-) diff --git a/nix-builder/lib/extension/torch/arch.nix b/nix-builder/lib/extension/torch/arch.nix index f9f5955e..d2c3c82e 100644 --- a/nix-builder/lib/extension/torch/arch.nix +++ b/nix-builder/lib/extension/torch/arch.nix @@ -244,17 +244,16 @@ stdenv.mkDerivation (prevAttrs: { ] ++ extraDeps; - env = - { - inherit variant; - } - // lib.optionalAttrs cudaSupport { - CUDAToolkit_ROOT = "${lib.getDev cudaPackages.cuda_nvcc}"; - } - // lib.optionalAttrs xpuSupport { - MKLROOT = oneapi-torch-dev; - SYCL_ROOT = oneapi-torch-dev; - }; + env = { + inherit variant; + } + // lib.optionalAttrs cudaSupport { + CUDAToolkit_ROOT = "${lib.getDev cudaPackages.cuda_nvcc}"; + } + // lib.optionalAttrs xpuSupport { + MKLROOT = oneapi-torch-dev; + SYCL_ROOT = oneapi-torch-dev; + }; # If we use the default setup, CMAKE_CUDA_HOST_COMPILER gets set to nixpkgs g++. dontSetupCUDAToolkitCompilers = true; diff --git a/nix-builder/lib/extension/tvm-ffi/arch.nix b/nix-builder/lib/extension/tvm-ffi/arch.nix index de5a58dd..c4b39c6a 100644 --- a/nix-builder/lib/extension/tvm-ffi/arch.nix +++ b/nix-builder/lib/extension/tvm-ffi/arch.nix @@ -236,17 +236,16 @@ stdenv.mkDerivation (prevAttrs: { ] ++ extraDeps; - env = - { - inherit variant; - } - // lib.optionalAttrs cudaSupport { - CUDAToolkit_ROOT = "${lib.getDev cudaPackages.cuda_nvcc}"; - } - // lib.optionalAttrs xpuSupport { - MKLROOT = oneapi-torch-dev; - SYCL_ROOT = oneapi-torch-dev; - }; + env = { + inherit variant; + } + // lib.optionalAttrs cudaSupport { + CUDAToolkit_ROOT = "${lib.getDev cudaPackages.cuda_nvcc}"; + } + // lib.optionalAttrs xpuSupport { + MKLROOT = oneapi-torch-dev; + SYCL_ROOT = oneapi-torch-dev; + }; # If we use the default setup, CMAKE_CUDA_HOST_COMPILER gets set to nixpkgs g++. dontSetupCUDAToolkitCompilers = true;