Commit 80e0143e authored by Edward Z. Yang's avatar Edward Z. Yang Committed by mcarilli
Browse files

Add missing semicolon. (#390)


Signed-off-by: default avatarEdward Z. Yang <ezyang@fb.com>
parent 665b2dd7
...@@ -9,7 +9,7 @@ void fused_adam_cuda(at::Tensor & p, at::Tensor & p_copy, at::Tensor & m, at::Te ...@@ -9,7 +9,7 @@ void fused_adam_cuda(at::Tensor & p, at::Tensor & p_copy, at::Tensor & m, at::Te
// C++ interface // C++ interface
void adam(at::Tensor & p, at::Tensor & p_copy, at::Tensor & m, at::Tensor & v, at::Tensor & g, float lr, float beta1, float beta2, float eps, float grad_scale, int step, int mode, int bias_correction, float decay) { void adam(at::Tensor & p, at::Tensor & p_copy, at::Tensor & m, at::Tensor & v, at::Tensor & g, float lr, float beta1, float beta2, float eps, float grad_scale, int step, int mode, int bias_correction, float decay) {
CHECK_INPUT(p) CHECK_INPUT(p);
if (p_copy.numel() > 0) CHECK_INPUT(p_copy); if (p_copy.numel() > 0) CHECK_INPUT(p_copy);
CHECK_INPUT(m); CHECK_INPUT(m);
CHECK_INPUT(v); CHECK_INPUT(v);
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment