Revert D24714803: make duplicate def() calls an error in the dispatcher. Updating all fb operators to use the new dispatcher registration API

Differential Revision:
D24714803

Original commit changeset: c809aad8a698

fbshipit-source-id: fb2ada65f9fc00d965708d202bd9d050f13ef467

Author: bdhirsh

nestedtensor/csrc/mha.cpp

@@ -62,9 +62,8 @@ at::Tensor min_mha(
   return attn_output;
 }
 
-TORCH_LIBRARY_FRAGMENT(nestedtensor, m) {
-    m.def("min_mha", min_mha);
-}
+static auto registry =
+    torch::RegisterOperators().op("nestedtensor::min_mha", &min_mha);
 
 } // namespace nested_tensor
 } // namespace torch

nestedtensor/csrc/py_init.cpp

@@ -145,34 +145,36 @@ inline std::vector<std::string> split_str(
   result.push_back(s);
   return result;
 }
-TORCH_LIBRARY_FRAGMENT(nestedtensor, m) {
-    m.def("is_nested_tensor_impl",
-            [](Tensor tensor) { return is_nested_tensor_impl(tensor); });
-    m.def("nested_dim",
+
+static auto registry =
+    torch::RegisterOperators()
+        .op("nestedtensor::is_nested_tensor_impl",
+            [](Tensor tensor) { return is_nested_tensor_impl(tensor); })
+        .op("nestedtensor::nested_dim",
             [](Tensor tensor) {
               return get_nested_tensor_impl(tensor)->nested_dim();
-            });
-    m.def("stack",
+            })
+        .op("nestedtensor::stack",
             [](std::vector<Tensor> tensors, int64_t dim) {
               return at::stack(TensorList(tensors), dim);
-            });
-    m.def("cat",
+            })
+        .op("nestedtensor::cat",
             [](std::vector<Tensor> tensors, int64_t dim) {
               return at::cat(TensorList(tensors), dim);
-            });
-    m.def("to_nested_tensor",
+            })
+        .op("nestedtensor::to_nested_tensor",
             [](Tensor tensor, c10::optional<int64_t> dim) {
               return get_nested_tensor_impl(tensor)->to_nested_tensor(dim);
-            });
-    m.def("sizes",
+            })
+        .op("nestedtensor::sizes",
             [](Tensor tensor) {
               return get_nested_tensor_impl(tensor)->opt_sizes();
-            });
-    m.def("len",
+            })
+        .op("nestedtensor::len",
             [](Tensor self) {
               return (int64_t)(get_nested_tensor_structure(self).degree());
-            });
-    m.def("str", [](Tensor tensor) {
+            })
+        .op("nestedtensor::str", [](Tensor tensor) {
           auto node = get_nested_tensor_structure(tensor);
           return NestedNode___str__(
               node,
@@ -201,8 +203,6 @@ TORCH_LIBRARY_FRAGMENT(nestedtensor, m) {
                 return result;
               });
         });
-}
-
 } // namespace
 } // namespace nested_tensor
 } // namespace torch

nestedtensor/csrc/totensor.cpp

@@ -109,11 +109,10 @@ Tensor NestedTensor_to_tensor(Tensor tensor, c10::optional<int64_t> dim_) {
   // return wrap_tensor_node(TensorNode(std::move(result)));
 }
 
-TORCH_LIBRARY_FRAGMENT(nestedtensor, m) {
-    m.def("to_tensor",
+static auto registry = torch::RegisterOperators().op(
+    "nestedtensor::to_tensor",
     [](Tensor tensor, c10::optional<int64_t> dim) {
       return NestedTensor_to_tensor(tensor, dim);
     });
-}
 
 } // namespace at