Fix some compilation issues.

huggingface · Jan 25, 2025 · 1b55287 · 1b55287
1 parent cb29411
commit 1b55287
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/candle-metal-kernels/src/scaled_dot_product_attention.metal b/candle-metal-kernels/src/scaled_dot_product_attention.metal
@@ -1404,7 +1404,7 @@ instantiate_fast_inference_self_attention_kernel(half, half, 16, 16, 256, 2, 2);
       const constant size_t& v_stride,                                       \
       const constant float& scale,                                           \
       const constant float& softcapping,                                     \
-      const device bool* mask [[function_constant(sdpa_vector_has_mask)]],,              \
+      const device bool* mask [[function_constant(sdpa_vector_has_mask)]],              \
       const constant int& mask_seq_stride [[function_constant(sdpa_vector_has_mask)]],   \
       const constant int& mask_head_stride [[function_constant(sdpa_vector_has_mask)]],  \
       uint3 tid [[threadgroup_position_in_grid]],                            \
@@ -1424,7 +1424,7 @@ instantiate_fast_inference_self_attention_kernel(half, half, 16, 16, 256, 2, 2);
       const constant size_t& v_stride,                                       \
       const constant float& scale,                                           \
       const constant float& softcapping,                                     \
-      const device bool* mask [[function_constant(sdpa_vector_has_mask)]],,              \
+      const device bool* mask [[function_constant(sdpa_vector_has_mask)]],              \
       const constant int& mask_seq_stride [[function_constant(sdpa_vector_has_mask)]],   \
       const constant int& mask_head_stride [[function_constant(sdpa_vector_has_mask)]],  \
       uint3 tid [[threadgroup_position_in_grid]],                            \