diff options
| author | Laurent Mazare <laurent.mazare@gmail.com> | 2023-07-23 12:31:17 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-07-23 11:31:17 +0100 |
| commit | e449ce53a2f3c85f23ca0f2e7d557a0d0003e0ca (patch) | |
| tree | b908de30c669bbbf20edb1604caec3450f76a01d /candle-examples/examples/custom-ops/cuda_kernels.rs | |
| parent | b8a10425ad550b04ccf3b5ff2493714615d7df4b (diff) | |
| download | candle-e449ce53a2f3c85f23ca0f2e7d557a0d0003e0ca.tar.gz candle-e449ce53a2f3c85f23ca0f2e7d557a0d0003e0ca.tar.bz2 candle-e449ce53a2f3c85f23ca0f2e7d557a0d0003e0ca.zip | |
Wrapping code to call the custom op. (#225)
* Wrapping code to call the custom op.
* Get the rms example to work.
* Get around rustfmt failing in the CI.
* Fix the rms computation.
Diffstat (limited to 'candle-examples/examples/custom-ops/cuda_kernels.rs')
| -rw-r--r-- | candle-examples/examples/custom-ops/cuda_kernels.rs | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/candle-examples/examples/custom-ops/cuda_kernels.rs b/candle-examples/examples/custom-ops/cuda_kernels.rs index 07d18342..0bee73aa 100644 --- a/candle-examples/examples/custom-ops/cuda_kernels.rs +++ b/candle-examples/examples/custom-ops/cuda_kernels.rs @@ -1 +1,2 @@ +#[rustfmt::skip] pub const LAYERNORM_KERNELS: &str = include_str!(concat!(env!("OUT_DIR"), "/examples/custom-ops/kernels//layernorm_kernels.ptx")); |