std :: ml::gpu_mps
Macro MPS_FP16
#define MPS_FP16 1Element types of a buffer (MpsBuf.dtype) and of the attention ops' dtype argument: 0 = fp32, or raw 16-bit values.
same values as gpu_cuda.h's CUDA_FP16 / CUDA_BF16
std :: ml::gpu_mps
#define MPS_FP16 1Element types of a buffer (MpsBuf.dtype) and of the attention ops' dtype argument: 0 = fp32, or raw 16-bit values.
same values as gpu_cuda.h's CUDA_FP16 / CUDA_BF16