flake.nix

{
  description = "Flake for Triton layer norm kernels";

  inputs = {
    kernel-builder.url = "github:huggingface/kernel-builder";
  };

  outputs =
    {
      self,
      kernel-builder,
    }:
    kernel-builder.lib.genFlakeOutputs {
      path = ./.;
      rev = self.shortRev or self.dirtyShortRev or self.lastModifiedDate;
      # Import-time autotune.
      doGetKernelCheck = false;
      pythonCheckInputs = pkgs: with pkgs; [ einops ];
    };
}