std :: ml::gpu_rocm
#include <std/ml/gpu_rocm.h> source ยท definition
#include <std/ml/gpu_rocm.h>
int rocm_sgd(struct RocmBuf w, struct RocmBuf g, float lr)
w -= lr * g, in place