// File: yuang-chen[P]paraGCN/src/optim.cpp
// Function: Adam::step()
void Adam::step() {
    step_count++;
    float step_size = params.lr * sqrtf(1 - powf(params.beta2, step_count)) / (1 - powf(params.beta1, step_count));
    for(auto &var: adamVars) {
        for(int i = 0; i < var.size(); i++) {
            float grad = (*var.grad)[i];
            if(var.decay) {
                grad += params.weight_decay * (*var.data)[i];
            }
            var.m[i] = params.beta1 * var.m[i] + (1.0 - params.beta1) * grad;
            var.v[i] = params.beta2 * var.v[i] + (1.0 - params.beta2) * grad * grad;
            (*var.data)[i] -= step_size * var.m[i] / (sqrtf(var.v[i]) + params.eps);
        }
    }
}

// Assembly:
pushq %rbp
movq %rsp, %rbp
subq $0x70, %rsp
movq %rdi, -0x8(%rbp)
movq -0x8(%rbp), %rax
movq %rax, -0x48(%rbp)
movl 0x14(%rax), %ecx
addl $0x1, %ecx
movl %ecx, 0x14(%rax)
movss (%rax), %xmm0
movss %xmm0, -0x4c(%rbp)
movss 0x8(%rax), %xmm0
cvtsi2ssl 0x14(%rax), %xmm1
callq 0x3390
movaps %xmm0, %xmm1
movss 0x271c(%rip), %xmm0     # 0x101cc
subss %xmm1, %xmm0
callq 0x3450
movq -0x48(%rbp), %rax
movaps %xmm0, %xmm1
movss -0x4c(%rbp), %xmm0
mulss %xmm1, %xmm0
movss %xmm0, -0x3c(%rbp)
movss 0x4(%rax), %xmm0
cvtsi2ssl 0x14(%rax), %xmm1
callq 0x3390
movq -0x48(%rbp), %rax
movaps %xmm0, %xmm2
movss -0x3c(%rbp), %xmm0
movss 0x26db(%rip), %xmm1     # 0x101cc
subss %xmm2, %xmm1
divss %xmm1, %xmm0
movss %xmm0, -0xc(%rbp)
addq $0x18, %rax
movq %rax, -0x18(%rbp)
movq -0x18(%rbp), %rdi
callq 0xdf50
movq %rax, -0x20(%rbp)
movq -0x18(%rbp), %rdi
callq 0xdf80
movq %rax, -0x28(%rbp)
leaq -0x20(%rbp), %rdi
leaq -0x28(%rbp), %rsi
callq 0xdfb0
testb $0x1, %al
jne 0xdb36
jmp 0xdd67
leaq -0x20(%rbp), %rdi
callq 0xdff0
movq %rax, -0x30(%rbp)
movl $0x0, -0x34(%rbp)
movl -0x34(%rbp), %eax
movl %eax, -0x50(%rbp)
movq -0x30(%rbp), %rdi
callq 0xd960
movl %eax, %ecx
movl -0x50(%rbp), %eax
cmpl %ecx, %eax
jge 0xdd57
movq -0x30(%rbp), %rax
movq 0x8(%rax), %rdi
movslq -0x34(%rbp), %rsi
callq 0x5520
movss (%rax), %xmm0
movss %xmm0, -0x38(%rbp)
movq -0x30(%rbp), %rax
testb $0x1, 0x40(%rax)
je 0xdbc3
movq -0x48(%rbp), %rax
movss 0x10(%rax), %xmm0
movss %xmm0, -0x54(%rbp)
movq -0x30(%rbp), %rax
movq (%rax), %rdi
movslq -0x34(%rbp), %rsi
callq 0x5520
movss -0x54(%rbp), %xmm0
movss (%rax), %xmm2
movss -0x38(%rbp), %xmm1
mulss %xmm2, %xmm0
addss %xmm1, %xmm0
movss %xmm0, -0x38(%rbp)
movq -0x48(%rbp), %rax
movss 0x4(%rax), %xmm0
movss %xmm0, -0x70(%rbp)
movq -0x30(%rbp), %rdi
addq $0x10, %rdi
movslq -0x34(%rbp), %rsi
callq 0x5520
movss -0x70(%rbp), %xmm0
movq %rax, %rcx
movq -0x48(%rbp), %rax
mulss (%rcx), %xmm0
cvtss2sd %xmm0, %xmm1
movss 0x4(%rax), %xmm0
cvtss2sd %xmm0, %xmm2
movsd 0x25b9(%rip), %xmm0     # 0x101c0
subsd %xmm2, %xmm0
movss -0x38(%rbp), %xmm2
cvtss2sd %xmm2, %xmm2
mulsd %xmm2, %xmm0
addsd %xmm1, %xmm0
cvtsd2ss %xmm0, %xmm0
movss %xmm0, -0x6c(%rbp)
movq -0x30(%rbp), %rdi
addq $0x10, %rdi
movslq -0x34(%rbp), %rsi
callq 0x5520
movss -0x6c(%rbp), %xmm0
movq %rax, %rcx
movq -0x48(%rbp), %rax
movss %xmm0, (%rcx)
movss 0x8(%rax), %xmm0
movss %xmm0, -0x68(%rbp)
movq -0x30(%rbp), %rdi
addq $0x28, %rdi
movslq -0x34(%rbp), %rsi
callq 0x5520
movss -0x68(%rbp), %xmm0
movq %rax, %rcx
movq -0x48(%rbp), %rax
mulss (%rcx), %xmm0
cvtss2sd %xmm0, %xmm1
movss 0x8(%rax), %xmm0
cvtss2sd %xmm0, %xmm2
movsd 0x253a(%rip), %xmm0     # 0x101c0
subsd %xmm2, %xmm0
movss -0x38(%rbp), %xmm2
cvtss2sd %xmm2, %xmm2
mulsd %xmm2, %xmm0
movss -0x38(%rbp), %xmm2
cvtss2sd %xmm2, %xmm2
mulsd %xmm2, %xmm0
addsd %xmm1, %xmm0
cvtsd2ss %xmm0, %xmm0
movss %xmm0, -0x64(%rbp)
movq -0x30(%rbp), %rdi
addq $0x28, %rdi
movslq -0x34(%rbp), %rsi
callq 0x5520
movss -0x64(%rbp), %xmm0
movss %xmm0, (%rax)
movss -0xc(%rbp), %xmm0
movss %xmm0, -0x60(%rbp)
movq -0x30(%rbp), %rdi
addq $0x10, %rdi
movslq -0x34(%rbp), %rsi
callq 0x5520
movss -0x60(%rbp), %xmm0
mulss (%rax), %xmm0
movss %xmm0, -0x5c(%rbp)
movq -0x30(%rbp), %rdi
addq $0x28, %rdi
movslq -0x34(%rbp), %rsi
callq 0x5520
movss (%rax), %xmm0
callq 0x3450
movq -0x48(%rbp), %rax
movaps %xmm0, %xmm1
movss -0x5c(%rbp), %xmm0
addss 0xc(%rax), %xmm1
divss %xmm1, %xmm0
movss %xmm0, -0x58(%rbp)
movq -0x30(%rbp), %rax
movq (%rax), %rdi
movslq -0x34(%rbp), %rsi
callq 0x5520
movss -0x58(%rbp), %xmm1
movss (%rax), %xmm0
subss %xmm1, %xmm0
movss %xmm0, (%rax)
movl -0x34(%rbp), %eax
addl $0x1, %eax
movl %eax, -0x34(%rbp)
jmp 0xdb4a
jmp 0xdd59
leaq -0x20(%rbp), %rdi
callq 0xe010
jmp 0xdb20
addq $0x70, %rsp
popq %rbp
retq
nopl (%rax)