Skip to content

Commit 7d7b505

Browse files
adarshxsadarshxs
authored
Merge branch 'main' into deepgemm-upd
2 parents 7c39dd9 + 2567cd6 commit 7d7b505

6 files changed

Lines changed: 33 additions & 30 deletions

File tree

‎flash-attn4/flake.lock‎

Lines changed: 10 additions & 9 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

‎flash-attn4/flake.nix‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
description = "Flake for Flash Attention 4 kernels";
33

44
inputs = {
5-
kernel-builder.url = "github:huggingface/kernels";
5+
kernel-builder.url = "github:huggingface/kernels/torch-2.12";
66
};
77

88
outputs =

‎paged-attention/flake.lock‎

Lines changed: 10 additions & 9 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

‎paged-attention/flake.nix‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
description = "Flake for attention kernels";
33

44
inputs = {
5-
kernel-builder.url = "github:huggingface/kernels";
5+
kernel-builder.url = "github:huggingface/kernels/torch-2.12";
66
};
77

88
outputs =

‎relu/flake.lock‎

Lines changed: 10 additions & 9 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

‎relu/flake.nix‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
description = "Flake for ReLU kernel";
33

44
inputs = {
5-
kernel-builder.url = "github:huggingface/kernels";
5+
kernel-builder.url = "github:huggingface/kernels/torch-2.12";
66
};
77

88
outputs =

0 commit comments

Comments
 (0)