std1.0.0-beta-7 SafeC docs

std :: ml::gpu_rocm

Function std::rocm_half_axpy

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

int rocm_half_axpy(
    struct RocmHalfBuf a,
    float k,
    struct RocmHalfBuf b,
    struct RocmHalfBuf out,
)

a + k b