File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 11{
22 description = "Kernels" ;
33
4+ nixConfig = {
5+ extra-substituters = [ "https://huggingface.cachix.org" ] ;
6+ extra-trusted-public-keys = [
7+ "huggingface.cachix.org-1:ynTPbLS0W8ofXd9fDjk1KvoFky9K2jhxe6r4nXAkc/o="
8+ ] ;
9+ } ;
10+
411 inputs = {
512 flake-utils . url = "github:numtide/flake-utils" ;
613 nixpkgs . url = "github:NixOS/nixpkgs/nixos-unstable-small" ;
121128
122129 buildSets = defaultBuildSetsPerSystem . ${ system } ;
123130 buildSetsByBackend = ( partitionBuildSetsBySystemBackend defaultBuildSets ) . ${ system } ;
124- buildSet = builtins . head buildSetsByBackend . cuda ;
131+ defaultBackend = if system == "aarch64-darwin" then "metal" else "cuda" ;
132+ buildSet = builtins . head buildSetsByBackend . ${ defaultBackend } ;
125133
126134 # Dev shells per framework.
127135 devShellByBackend = lib . mapAttrs (
You can’t perform that action at this time.
0 commit comments