Hugging Face's logo

Kernels:
kernels-community
/
causal-conv1d

Kernel card Files
xet
 Community
kernels-bot commited on
Commit
a3baa8b
·
verified ·
1 Parent(s): eb9db37

Uploaded using `kernel-builder`.

Browse files
Files changed (1) hide show
  1. README.md +17 -12
README.md CHANGED
@@ -1,22 +1,27 @@
1
  ---
 
2
  license: bsd-3-clause
3
- tags:
4
- - kernels
5
  ---
6
 
7
- ## causal-conv1d
8
 
9
- Causal [depthwise conv1d kernel](https://github.com/Dao-AILab/causal-conv1d/) by Tri Dao.
10
 
11
- Kernel source: https://github.com/huggingface/kernels-community/tree/main/causal-conv1d
 
 
12
 
13
- ### Performance
 
14
 
15
- <img class="dark:hidden border border-gray-200 dark:border-gray-700 rounded-lg" src="media/benches_light_animation.svg" />
16
- <img class="hidden dark:block border border-gray-200 dark:border-gray-700 rounded-lg" src="media/benches_dark_animation.svg" />
17
 
18
- <img class="dark:hidden border border-gray-200 dark:border-gray-700 rounded-lg" src="media/benches_light_latency.svg" />
19
- <img class="hidden dark:block border border-gray-200 dark:border-gray-700 rounded-lg" src="media/benches_dark_latency.svg" />
 
 
20
 
21
- <img class="dark:hidden border border-gray-200 dark:border-gray-700 rounded-lg" src="media/benches_light_throughput.svg" />
22
- <img class="hidden dark:block border border-gray-200 dark:border-gray-700 rounded-lg" src="media/benches_dark_throughput.svg" />
 
 
1
  ---
2
+ library_name: kernels
3
  license: bsd-3-clause
 
 
4
  ---
5
 
6
+ This is the repository card of kernels-community/causal-conv1d that has been pushed on the Hub. It was built to be used with the [`kernels` library](https://github.com/huggingface/kernels). This card was automatically generated.
7
 
8
+ ## How to use
9
 
10
+ ```python
11
+ # make sure `kernels` is installed: `pip install -U kernels`
12
+ from kernels import get_kernel
13
 
14
+ kernel_module = get_kernel("kernels-community/causal-conv1d")
15
+ causal_conv1d_fn = kernel_module.causal_conv1d_fn
16
 
17
+ causal_conv1d_fn(...)
18
+ ```
19
 
20
+ ## Available functions
21
+ - `causal_conv1d_fn`
22
+ - `causal_conv1d_update`
23
+ - `causal_conv1d_varlen_states`
24
 
25
+ ## Benchmarks
26
+
27
+ Benchmarking script is available for this kernel. Run `kernels benchmark kernels-community/causal-conv1d`.