std :: ml::gpu_cuda
#include <std/ml/gpu_cuda.h> source ยท definition
#include <std/ml/gpu_cuda.h>
int cuda_sgd2( struct CudaBuf w1, struct CudaBuf g1, struct CudaBuf w2, struct CudaBuf g2, float lr, )
Both SGD updates of a two-tensor model in one launch: w1 -= lrg1, w2 -= lrg2.