[autoparallel] add experimental view handler (#2011)

* [autoparallel] add experimental view handler

* polish

* polish

* polish code

* rename variables
This commit is contained in:
YuliangLiu0306
2022-11-24 11:34:41 +08:00
committed by GitHub
parent d655eea515
commit 1438993113
8 changed files with 459 additions and 2 deletions

View File

@@ -51,6 +51,8 @@ class OperationData:
"""
if isinstance(data, torch.Tensor):
return data.shape
elif isinstance(data, torch.Size):
return None
elif isinstance(data, (tuple, list)):
data_type = type(data)
return data_type([_infer_logical_shape(d) for d in data])