huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-17 11:08:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add a cuda kernel for embeddings.

Browse Source
This commit is contained in:
laurent
2023-06-26 11:47:57 +01:00
parent 5952c3fa91
commit 16f0f5b9d2
5 changed files with 101 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
kernels/src/lib.rs
View File

@ -1,4 +1,5 @@
pub const AFFINE: &str = include_str!(concat!(env!("OUT_DIR"), "/affine.ptx"));
pub const BINARY: &str = include_str!(concat!(env!("OUT_DIR"), "/binary.ptx"));
pub const EMBEDDINGS: &str = include_str!(concat!(env!("OUT_DIR"), "/embeddings.ptx"));
pub const FILL: &str = include_str!(concat!(env!("OUT_DIR"), "/fill.ptx"));
pub const UNARY: &str = include_str!(concat!(env!("OUT_DIR"), "/unary.ptx"));