std :: ml::gpu_rocm
Function std::rocm_half_sgd2
int rocm_half_sgd2(
struct RocmHalfBuf w1,
struct RocmHalfBuf g1,
struct RocmHalfBuf w2,
struct RocmHalfBuf g2,
float lr,
)rocm_half_sgd on two weight/gradient pairs in one launch.