diff options
Diffstat (limited to 'candle-flash-attn/src/ffi.rs')
| -rw-r--r-- | candle-flash-attn/src/ffi.rs | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/candle-flash-attn/src/ffi.rs b/candle-flash-attn/src/ffi.rs index f4415539..ae61c405 100644 --- a/candle-flash-attn/src/ffi.rs +++ b/candle-flash-attn/src/ffi.rs @@ -7,6 +7,8 @@ extern "C" { v_ptr: *const c_void, o_ptr: *const c_void, softmax_lse_ptr: *const c_void, + cu_seqlens_q_ptr: *const i32, + cu_seqlens_k_ptr: *const i32, q_batch_stride: u32, k_batch_stride: u32, |