Kernels:

amd
/

aiter-flash-attn

Trusted publisher

kernels-bot commited on Sep 24

Commit

35f71ff

verified ·

1 Parent(s): d8a466b

Uploaded using `kernel-builder`.

Files changed (1) hide show

README.md CHANGED Viewed

@@ -11,7 +11,7 @@ This is the repository card of kernels-community/aiter-flash-attn that has been
 # make sure `kernels` is installed: `pip install -U kernels`
 from kernels import get_kernel
-kernel_module = get_kernel("kernels-community/aiter-flash-attn", version=1)
 __kernel_metadata__ = kernel_module.__kernel_metadata__
 __kernel_metadata__(...)

 # make sure `kernels` is installed: `pip install -U kernels`
 from kernels import get_kernel
+kernel_module = get_kernel("kernels-community/aiter-flash-attn", version=2)
 __kernel_metadata__ = kernel_module.__kernel_metadata__
 __kernel_metadata__(...)