|
@@ -126,16 +126,9 @@ let
|
|
|
++ optionals useMetalKit [ MetalKit ];
|
|
++ optionals useMetalKit [ MetalKit ];
|
|
|
|
|
|
|
|
cudaBuildInputs = with cudaPackages; [
|
|
cudaBuildInputs = with cudaPackages; [
|
|
|
- cuda_cccl.dev # <nv/target>
|
|
|
|
|
-
|
|
|
|
|
- # A temporary hack for reducing the closure size, remove once cudaPackages
|
|
|
|
|
- # have stopped using lndir: https://github.com/NixOS/nixpkgs/issues/271792
|
|
|
|
|
- cuda_cudart.dev
|
|
|
|
|
- cuda_cudart.lib
|
|
|
|
|
- cuda_cudart.static
|
|
|
|
|
- libcublas.dev
|
|
|
|
|
- libcublas.lib
|
|
|
|
|
- libcublas.static
|
|
|
|
|
|
|
+ cuda_cudart
|
|
|
|
|
+ cuda_cccl # <nv/target>
|
|
|
|
|
+ libcublas
|
|
|
];
|
|
];
|
|
|
|
|
|
|
|
rocmBuildInputs = with rocmPackages; [
|
|
rocmBuildInputs = with rocmPackages; [
|