We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
get_kernel
1 parent 6dfe59c commit 34eb665Copy full SHA for 34eb665
1 file changed
kernels/README.md
@@ -30,7 +30,7 @@ import torch
30
from kernels import get_kernel
31
32
# Download optimized kernels from the Hugging Face hub
33
-activation = get_kernel("kernels-community/activation")
+activation = get_kernel("kernels-community/activation", version=1)
34
35
# Random tensor
36
x = torch.randn((10, 10), dtype=torch.float16, device="cuda")
0 commit comments