rustfmt

huggingface · Nov 28, 2024 · 3c82d21 · 3c82d21
1 parent 57312cd
commit 3c82d21
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 5 deletions.
diff --git a/candle-examples/examples/mamba-minimal/model.rs b/candle-examples/examples/mamba-minimal/model.rs
@@ -17,7 +17,7 @@ pub struct Config {
 impl Config {
     fn vocab_size(&self) -> usize {
         let pad = self.pad_vocab_size_multiple;
-        (self.vocab_size + pad - 1) / pad * pad
+        self.vocab_size.div_ceil(pad) * pad
     }
 
     fn dt_rank(&self) -> usize {

diff --git a/candle-metal-kernels/src/utils.rs b/candle-metal-kernels/src/utils.rs
@@ -184,7 +184,8 @@ impl<'a> AsRef<metal::ComputeCommandEncoderRef> for WrappedEncoder<'a> {
 }
 
 impl EncoderProvider for &metal::CommandBuffer {
-    type Encoder<'a> = WrappedEncoder<'a>
+    type Encoder<'a>
+        = WrappedEncoder<'a>
     where
         Self: 'a;
     fn encoder(&self) -> Self::Encoder<'_> {
@@ -196,7 +197,8 @@ impl EncoderProvider for &metal::CommandBuffer {
 }
 
 impl EncoderProvider for &metal::CommandBufferRef {
-    type Encoder<'a> = WrappedEncoder<'a>
+    type Encoder<'a>
+        = WrappedEncoder<'a>
     where
         Self: 'a;
     fn encoder(&self) -> Self::Encoder<'_> {
@@ -208,7 +210,8 @@ impl EncoderProvider for &metal::CommandBufferRef {
 }
 
 impl EncoderProvider for &ComputeCommandEncoderRef {
-    type Encoder<'a> = WrappedEncoder<'a>
+    type Encoder<'a>
+        = WrappedEncoder<'a>
     where
         Self: 'a;
     fn encoder(&self) -> Self::Encoder<'_> {

diff --git a/candle-nn/tests/sdpa.rs b/candle-nn/tests/sdpa.rs
@@ -115,7 +115,7 @@ mod metal_sdpa_tests {
         let error: f32 = ((&ground_truth - &sdpa_output)?.abs()? / &ground_truth.abs()?)?
             .sum_all()?
             .to_scalar()?;
-        
+
         assert!(error <= 0.0004, "{}", error);
 
         Ok(())