fix: refactors and adjust name

torch-ext/torch_binding.cpp

@@ -5,7 +5,7 @@
 
     
 TORCH_LIBRARY_EXPAND(TORCH_EXTENSION_NAME, ops) {
-  ops.def("adam_atan2("
+  ops.def("adam_atan2_cuda_impl_("
       "Tensor(a!)[] params, "
       "Tensor(b!)[] grads, "
       "Tensor(c!)[] exp_avgs, "

torch-ext/torch_binding.h

@@ -16,14 +16,3 @@ void adam_atan2_cuda_impl_(
     const double weight_decay);
 
 }
-
-// void adam_atan2_cuda_impl_(
-//     std::vector<at::Tensor, std::allocator<at::Tensor> > params, 
-//     std::vector<at::Tensor, std::allocator<at::Tensor> > grads, 
-//     std::vector<at::Tensor, std::allocator<at::Tensor> > exp_avgs, 
-//     std::vector<at::Tensor, std::allocator<at::Tensor> > exp_avg_sqs, 
-//     std::vector<at::Tensor, std::allocator<at::Tensor> > state_steps, 
-//     const double lr, 
-//     const double beta1, 
-//     const double beta2, 
-//     const double weight_decay);