std1.0.0-beta-7 SafeC docs

std :: ml::gpu_cuda_attention

Function std::cuda_linear_attention_ex

#include <std/ml/gpu_cuda_attention.h> source ยท definition

int cuda_linear_attention_ex(
    struct CudaBuf q,
    struct CudaBuf k,
    struct CudaBuf v,
    struct CudaBuf out,
    struct CudaAttnShape s,
    int causal,
)