skip adamw temporarily

DeepLink-org · Mar 20, 2024 · 86fd184 · 86fd184
1 parent 1e278b0
commit 86fd184
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/csrc/extensions.cpp b/csrc/extensions.cpp
@@ -367,9 +367,9 @@ PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {
   if (&diopiApplyPenalty != nullptr) {
     m.def("apply_penalty", &extApplyPenalty, "deeplink ext_apply_penalty");
   }
-  if (&diopiAdamW != nullptr) {
-    m.def("adamw", &extAdamW, "deeplink ext_adamw");
-  }
+  // if (&diopiAdamW != nullptr) {
+  //   m.def("adamw", &extAdamW, "deeplink ext_adamw");
+  // }
 }
 
 }  // namespace dipu::dipu_ext