Skip to content

Commit

Permalink
Fix BackendValues in backend-comparison after removal of jit suffix (#…
Browse files Browse the repository at this point in the history
  • Loading branch information
syl20bnr authored Jan 30, 2025
1 parent 3903172 commit 6d8dd69
Show file tree
Hide file tree
Showing 4 changed files with 12 additions and 12 deletions.
14 changes: 7 additions & 7 deletions backend-comparison/src/burnbenchapp/base.rs
Original file line number Diff line number Diff line change
Expand Up @@ -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")]
Expand All @@ -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)]
Expand Down
4 changes: 2 additions & 2 deletions crates/burn-remote/src/server/session.rs
Original file line number Diff line number Diff line change
Expand Up @@ -101,12 +101,12 @@ impl<B: ReprBackend> SessionManager<B> {

impl<B: ReprBackend> Session<B> {
fn new(runner: Runner<B>) -> 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),
}
}

Expand Down
4 changes: 2 additions & 2 deletions crates/burn-tensor/src/tensor/backend/conversion.rs
Original file line number Diff line number Diff line change
Expand Up @@ -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);
Expand All @@ -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);
Expand Down
2 changes: 1 addition & 1 deletion examples/wgan/src/model.rs
Original file line number Diff line number Diff line change
Expand Up @@ -83,7 +83,7 @@ impl<B: Backend> Discriminator<B> {
}
}

// 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
Expand Down

0 comments on commit 6d8dd69

Please sign in to comment.