kernels-community
/

adam-atan2

Model card Files Files and versions

drbh commited on 18 days ago

Commit

0709eaa

·

1 Parent(s): e612007

fix: adjust binding name

Files changed (1) hide show

torch-ext/torch_binding.cpp +1 -1

torch-ext/torch_binding.cpp CHANGED Viewed

@@ -15,7 +15,7 @@ TORCH_LIBRARY_EXPAND(TORCH_EXTENSION_NAME, ops) {
       "float beta1, "
       "float beta2, "
       "float weight_decay) -> ()");
-  ops.impl("adam_atan2", torch::kCUDA, &adam_atan2::adam_atan2_cuda_impl_);
 }
 REGISTER_EXTENSION(TORCH_EXTENSION_NAME)

       "float beta1, "
       "float beta2, "
       "float weight_decay) -> ()");
+  ops.impl("adam_atan2_cuda_impl_", torch::kCUDA, &adam_atan2::adam_atan2_cuda_impl_);
 }
 REGISTER_EXTENSION(TORCH_EXTENSION_NAME)