[autoparallel] added binary elementwise node handler (#1758)

* [autoparallel] added binary elementwise node handler

* polish code

colossalai/auto_parallel/tensor_shard/node_handler/registry.py

@@ -8,7 +8,12 @@ class Registry:
     def register(self, source):
 
         def wrapper(func):
-            self.store[source] = func
+            if isinstance(source, (list, tuple)):
+                # support register a list of items for this func
+                for element in source:
+                    self.store[element] = func
+            else:
+                self.store[source] = func
             return func
 
         return wrapper