diff --git a/dev/sycl/adamw.cpp b/dev/sycl/adamw.cpp index bc88a2f..c4da690 100644 --- a/dev/sycl/adamw.cpp +++ b/dev/sycl/adamw.cpp @@ -6,6 +6,7 @@ #include #include "common.hpp" +// -- void adamw_cpu(float* params_memory, const float* grads_memory, float* m_memory, float* v_memory, int t, long num_parameters, float learning_rate=1e-3, float beta1=0.9, float beta2=0.999, float eps=1e-8, float weight_decay=0.0) { for (long i = 0; i < num_parameters; i++) { @@ -192,4 +193,4 @@ int main(int argc, char** argv) { delete[] v_memory; return 0; -} \ No newline at end of file +}