huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-16 18:48:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

Kernel build example (#224)

Browse Source 
* Build example kernels.

* Add some sample custom kernel.

* Get the example kernel to compile.

* Add some cuda code.

* More cuda custom op.

* More cuda custom ops.
This commit is contained in:
Laurent Mazare
2023-07-23 08:15:37 +02:00
committed by GitHub
parent 5f20acf080
commit b8a10425ad
7 changed files with 427 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
candle-examples/examples/custom-ops/cuda_kernels.rs Normal file
View File

@ -0,0 +1 @@
pub const LAYERNORM_KERNELS: &str = include_str!(concat!(env!("OUT_DIR"), "/examples/custom-ops/kernels//layernorm_kernels.ptx"));