diff --git a/backend-comparison/src/burnbenchapp/base.rs b/backend-comparison/src/burnbenchapp/base.rs index 9eba1485b3..4fb31edab8 100644 --- a/backend-comparison/src/burnbenchapp/base.rs +++ b/backend-comparison/src/burnbenchapp/base.rs @@ -62,6 +62,13 @@ enum BackendValues { CandleCuda, #[strum(to_string = "candle-metal")] CandleMetal, + #[strum(to_string = "cuda")] + Cuda, + #[strum(to_string = "cuda-fusion")] + CudaFusion, + #[cfg(target_os = "linux")] + #[strum(to_string = "hip")] + Hip, #[strum(to_string = "ndarray")] Ndarray, #[strum(to_string = "ndarray-blas-accelerate")] @@ -82,13 +89,6 @@ enum BackendValues { WgpuSpirv, #[strum(to_string = "wgpu-spirv-fusion")] WgpuSpirvFusion, - #[strum(to_string = "cuda-jit")] - CudaJit, - #[strum(to_string = "cuda-jit-fusion")] - CudaJitFusion, - #[cfg(target_os = "linux")] - #[strum(to_string = "hip-jit")] - HipJit, } #[derive(Debug, Clone, PartialEq, Eq, ValueEnum, Display, EnumIter)] diff --git a/crates/burn-remote/src/server/session.rs b/crates/burn-remote/src/server/session.rs index 7d32d04b74..3da6b2afa1 100644 --- a/crates/burn-remote/src/server/session.rs +++ b/crates/burn-remote/src/server/session.rs @@ -101,12 +101,12 @@ impl SessionManager { impl Session { fn new(runner: Runner) -> Self { - let (sender, reveiver) = std::sync::mpsc::sync_channel(1); + let (sender, receiver) = std::sync::mpsc::sync_channel(1); Self { runner, streams: Default::default(), sender, - receiver: Some(reveiver), + receiver: Some(receiver), } } diff --git a/crates/burn-tensor/src/tensor/backend/conversion.rs b/crates/burn-tensor/src/tensor/backend/conversion.rs index 46b0423b71..6aebe06463 100644 --- a/crates/burn-tensor/src/tensor/backend/conversion.rs +++ b/crates/burn-tensor/src/tensor/backend/conversion.rs @@ -188,7 +188,7 @@ mod tests { } #[test] - fn should_build_indices_2d_complexe() { + fn should_build_indices_2d_complex() { let shape = Shape::new([2, 3]); let indices = build_indices(&shape, Order::Left); @@ -206,7 +206,7 @@ mod tests { } #[test] - fn should_build_indices_3d_complexe() { + fn should_build_indices_3d_complex() { let shape = Shape::new([2, 5, 3]); let indices = build_indices(&shape, Order::Left); diff --git a/examples/wgan/src/model.rs b/examples/wgan/src/model.rs index b9615f5270..755d8e9e1d 100644 --- a/examples/wgan/src/model.rs +++ b/examples/wgan/src/model.rs @@ -83,7 +83,7 @@ impl Discriminator { } } -// Use model config to construct a generative and adverserial model +// Use model config to construct a generative and adversarial model #[derive(Config, Debug)] pub struct ModelConfig { /// Dimensionality of the latent space