std1.0.0-beta-7 SafeC docs

std :: ml::gpu_webgpu

Function std::webgpu_matmul

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

int webgpu_matmul(
    struct WebGpuBuf a,
    struct WebGpuBuf b,
    struct WebGpuBuf out,
    unsigned long M,
    unsigned long K,
    unsigned long N,
)

out [M, N] = a [M, K] . b [K, N], row-major. a and b share a dtype (fp32, fp16 or bf16; products accumulate in fp32), out is any dtype. Queued.