[hotfix] make Gemini work for conv DNN (#1998)

colossalai/nn/_ops/element_wise.py

 import torch
 import torch.nn.functional as F
 from torch import Tensor
-from colossalai.tensor.op_wrapper import colo_op_impl
+
 from colossalai.tensor import ColoTensor, ColoTensorSpec
-from ._utils import GeneralTensor
+from colossalai.tensor.op_wrapper import colo_op_impl
+
+from ._utils import GeneralTensor, convert_to_colo_tensor
 
 
 def register_elementwise_op(op):
@@ -15,8 +17,13 @@ def register_elementwise_op(op):
         as ``torch.nn.functional.gelu`` or ``torch.nn.functional.relu``.
         This method computes on either a normal tensor or a sharded tensor.
         """
-
+        if 'inplace' in kwargs:
+            # TODO(jiaruifang) inplace will cause bugs
+            input_tensor = input_tensor.clone()
+            return op(input_tensor, *args, **kwargs)
+        else:
             output = op(input_tensor, *args, **kwargs)
+            # return output
             if isinstance(input_tensor, ColoTensor):
                 if isinstance(output, str):
                     return output