diff options
Diffstat (limited to 'candle-examples/examples/custom-ops')
| -rw-r--r-- | candle-examples/examples/custom-ops/cuda_kernels.rs | 3 | ||||
| -rw-r--r-- | candle-examples/examples/custom-ops/main.rs | 3 |
2 files changed, 3 insertions, 3 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")); diff --git a/candle-examples/examples/custom-ops/main.rs b/candle-examples/examples/custom-ops/main.rs index f2f534dc..30e413c1 100644 --- a/candle-examples/examples/custom-ops/main.rs +++ b/candle-examples/examples/custom-ops/main.rs @@ -6,7 +6,8 @@ #[cfg(feature = "mkl")] extern crate intel_mkl_src; -#[allow(unused)] +#[rustfmt::skip] +#[cfg(feature = "cuda")] mod cuda_kernels; use clap::Parser; |