flake.nix

{
  description = "Flake for Torch kernel extension";
  inputs = {
    # Go back to main after https://github.com/huggingface/kernel-builder/pull/217
    # is merged.
    kernel-builder.url = "github:huggingface/kernel-builder/fix-cuda-test-shells";
  };
  outputs = { self, kernel-builder, }:
    kernel-builder.lib.genFlakeOutputs {
      path = ./.;
      rev = self.shortRev or self.dirtyShortRev or self.lastModifiedDate;
    };
}