| { | |
| description = "Flake for quantization kernels"; | |
| inputs = { | |
| flake-utils.url = "github:numtide/flake-utils"; | |
| nixpkgs.follows = "kernel-builder/nixpkgs"; | |
| kernel-builder.url = "path:/home/daniel/git/kernel-builder"; | |
| }; | |
| outputs = | |
| { | |
| self, | |
| flake-utils, | |
| nixpkgs, | |
| kernel-builder, | |
| }: | |
| flake-utils.lib.eachSystem [ flake-utils.lib.system.x86_64-linux ] (system: { | |
| devShells = rec { | |
| default = shells.torch24-cxx98-cu124-x86_64-linux; | |
| shells = kernel-builder.lib.${system}.torchExtensionShells ./.; | |
| }; | |
| packages = { | |
| bundle = kernel-builder.lib.${system}.buildTorchExtensionBundle ./.; | |
| redistributable = kernel-builder.lib.${system}.buildDistTorchExtensions ./.; | |
| }; | |
| }); | |
| } | |