flash-attn / flake.nix
drbh
fix: limit build head size and cuda caps for dev
9002ff5
{
description = "Flake for ReLU kernel";
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;
};
}