pub unsafe extern "C" fn cudnnSetAttnDescriptor(
attnDesc: cudnnAttnDescriptor_t,
attnMode: c_uint,
nHeads: c_int,
smScaler: f64,
dataType: cudnnDataType_t,
computePrec: cudnnDataType_t,
mathType: cudnnMathType_t,
attnDropoutDesc: cudnnDropoutDescriptor_t,
postDropoutDesc: cudnnDropoutDescriptor_t,
qSize: c_int,
kSize: c_int,
vSize: c_int,
qProjSize: c_int,
kProjSize: c_int,
vProjSize: c_int,
oProjSize: c_int,
qoMaxSeqLength: c_int,
kvMaxSeqLength: c_int,
maxBatchSize: c_int,
maxBeamSize: c_int
) -> cudnnStatus_t