[autoparallel] resnet block runtime apply (#1709)

* [autoparallel] resnet block runtime apply

* seperate buffer and parameter in MemoryCost

* polish code

* add comments and todos

* fix test issue

colossalai/auto_parallel/tensor_shard/sharding_strategy.py

@@ -20,7 +20,8 @@ class OperationDataType(Enum):
     INPUT = 0
     ARG = 1
     PARAM = 2
-    OUTPUT = 3
+    BUFFER = 3
+    OUTPUT = 4
 
 
 @dataclass
@@ -80,6 +81,7 @@ class MemoryCost:
     """
     activation: int = 0
     parameter: int = 0
+    buffer: int = 0
 
 
 @dataclass