std :: ml::gpu_rocm
Function std::rocm_half_sgd
int rocm_half_sgd(struct RocmHalfBuf w, struct RocmHalfBuf g, float lr)w = round(w - lr * g), computed in fp32, in place.
std :: ml::gpu_rocm
int rocm_half_sgd(struct RocmHalfBuf w, struct RocmHalfBuf g, float lr)w = round(w - lr * g), computed in fp32, in place.