Function std::cuda_attention_head_dim_ok

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

int cuda_attention_head_dim_ok(unsigned long headDim)

1 if 'headDim' is supported by the kernels below.