flake.nix

{
  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 ./.;
      };
    });
}