diff options
| author | Laurent Mazare <laurent.mazare@gmail.com> | 2024-01-07 17:18:46 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-01-07 17:18:46 +0100 |
| commit | 89b5a068585b73193d2004a7293d5b2fa6c30bfd (patch) | |
| tree | 1bb39bbcf60bf4140c704baa827c382693d3497e /candle-examples/examples/custom-ops/cuda_kernels.rs | |
| parent | 30313c308106fff7b20fc8cb2b27eb79800cb818 (diff) | |
| download | candle-89b5a068585b73193d2004a7293d5b2fa6c30bfd.tar.gz candle-89b5a068585b73193d2004a7293d5b2fa6c30bfd.tar.bz2 candle-89b5a068585b73193d2004a7293d5b2fa6c30bfd.zip | |
Use bindgen-cuda for the custom-kernel example. (#1536)
* Use bindgen-cuda for the custom-kernel example.
* Only depend on the kernels when cuda is enabled.
* Skip rustfmt.
Diffstat (limited to 'candle-examples/examples/custom-ops/cuda_kernels.rs')
| -rw-r--r-- | candle-examples/examples/custom-ops/cuda_kernels.rs | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/candle-examples/examples/custom-ops/cuda_kernels.rs b/candle-examples/examples/custom-ops/cuda_kernels.rs index 0bee73aa..c00b601b 100644 --- a/candle-examples/examples/custom-ops/cuda_kernels.rs +++ b/candle-examples/examples/custom-ops/cuda_kernels.rs @@ -1,2 +1 @@ -#[rustfmt::skip] -pub const LAYERNORM_KERNELS: &str = include_str!(concat!(env!("OUT_DIR"), "/examples/custom-ops/kernels//layernorm_kernels.ptx")); +pub const LAYERNORM_KERNELS: &str = include_str!(concat!(env!("OUT_DIR"), "/layernorm_kernels.ptx")); |