Format

huggingface · Jan 16, 2025 · 6fd50d8 · 6fd50d8
1 parent 8d26067
commit 6fd50d8
Showing 1 changed file with 12 additions and 9 deletions.
diff --git a/candle-metal-kernels/src/lib.rs b/candle-metal-kernels/src/lib.rs
@@ -1906,10 +1906,11 @@ pub fn call_sdpa_vector(
         alpha
     };
 
-    let constants = Some(ConstantValues::new(vec![
-        (20, Value::Bool(/* sdpa_vector_has_mask */ false)),
-    ]));
-
+    let constants = Some(ConstantValues::new(vec![(
+        20,
+        Value::Bool(/* sdpa_vector_has_mask */ false),
+    )]));
+
     let pipeline = kernels.load_pipeline_with_constants(device, Source::Sdpa, name, constants)?;
     let encoder = ep.encoder();
     let encoder: &ComputeCommandEncoderRef = encoder.as_ref();
@@ -2022,11 +2023,13 @@ pub fn call_sdpa_vector_2pass(
             alpha
         };
 
-        let constants = Some(ConstantValues::new(vec![
-            (20, Value::Bool(/* sdpa_vector_has_mask */ false)),
-        ]));
-
-        let pipeline = kernels.load_pipeline_with_constants(device, Source::Sdpa, &name_pass1, constants)?;
+        let constants = Some(ConstantValues::new(vec![(
+            20,
+            Value::Bool(/* sdpa_vector_has_mask */ false),
+        )]));
+
+        let pipeline =
+            kernels.load_pipeline_with_constants(device, Source::Sdpa, &name_pass1, constants)?;
         let encoder = ep.encoder();
         let encoder: &ComputeCommandEncoderRef = encoder.as_ref();
         encoder.set_compute_pipeline_state(&pipeline);