mirror of
https://github.com/huggingface/candle.git
synced 2025-06-16 10:38:54 +00:00
e449ce53a2
* Wrapping code to call the custom op. * Get the rms example to work. * Get around rustfmt failing in the CI. * Fix the rms computation.
3 lines
148 B
Rust
3 lines
148 B
Rust
#[rustfmt::skip]
|
|
pub const LAYERNORM_KERNELS: &str = include_str!(concat!(env!("OUT_DIR"), "/examples/custom-ops/kernels//layernorm_kernels.ptx"));
|