pub unsafe extern "C" fn cudnnMultiHeadAttnForward(
handle: cudnnHandle_t,
attnDesc: cudnnAttnDescriptor_t,
currIdx: c_int,
loWinIdx: *const c_int,
hiWinIdx: *const c_int,
devSeqLengthsQO: *const c_int,
devSeqLengthsKV: *const c_int,
qDesc: cudnnSeqDataDescriptor_t,
queries: *const c_void,
residuals: *const c_void,
kDesc: cudnnSeqDataDescriptor_t,
keys: *const c_void,
vDesc: cudnnSeqDataDescriptor_t,
values: *const c_void,
oDesc: cudnnSeqDataDescriptor_t,
out: *mut c_void,
weightSizeInBytes: usize,
weights: *const c_void,
workSpaceSizeInBytes: usize,
workSpace: *mut c_void,
reserveSpaceSizeInBytes: usize,
reserveSpace: *mut c_void
) -> cudnnStatus_t