danieldk HF staff commited on
Commit
b31d486
·
1 Parent(s): d156726

Make this a repo a flake for convenience

Browse files
Files changed (1) hide show
  1. flake.nix +29 -0
flake.nix ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ description = "Flake for quantization kernels";
3
+
4
+ inputs = {
5
+ flake-utils.url = "github:numtide/flake-utils";
6
+ nixpkgs.follows = "kernel-builder/nixpkgs";
7
+ kernel-builder.url = "path:/home/daniel/git/kernel-builder";
8
+ };
9
+
10
+ outputs =
11
+ {
12
+ self,
13
+ flake-utils,
14
+ nixpkgs,
15
+ kernel-builder,
16
+ }:
17
+ flake-utils.lib.eachSystem [ flake-utils.lib.system.x86_64-linux ] (system: {
18
+
19
+ devShells = rec {
20
+ default = shells.torch24-cxx98-cu124-x86_64-linux;
21
+ shells = kernel-builder.lib.${system}.torchExtensionShells ./.;
22
+ };
23
+
24
+ packages = {
25
+ bundle = kernel-builder.lib.${system}.buildTorchExtensionBundle ./.;
26
+ redistributable = kernel-builder.lib.${system}.buildDistTorchExtensions ./.;
27
+ };
28
+ });
29
+ }