digger yu 32f81f14d4 [NFC] fix typo colossalai/amp auto_parallel autochunk (#3756)		2023-05-19 13:50:00 +08:00
..
__init__.py	[autoparallel] Patch meta information for nodes that will not be handled by SPMD solver (#2823)	2023-02-22 10:28:56 +08:00
activation.py	[autoparallel]integrate auto parallel feature with new tracer (#3408)	2023-04-04 17:40:45 +08:00
binary_elementwise_ops.py	[autoparallel]integrate auto parallel feature with new tracer (#3408)	2023-04-04 17:40:45 +08:00
conv.py	[autoparallel]integrate auto parallel feature with new tracer (#3408)	2023-04-04 17:40:45 +08:00
embedding.py	[autoparallel]integrate auto parallel feature with new tracer (#3408)	2023-04-04 17:40:45 +08:00
linear.py	[NFC] fix typo colossalai/amp auto_parallel autochunk (#3756)	2023-05-19 13:50:00 +08:00
non_spmd.py	[autoparallel]integrate auto parallel feature with new tracer (#3408)	2023-04-04 17:40:45 +08:00
norm.py	[autoparallel]integrate auto parallel feature with new tracer (#3408)	2023-04-04 17:40:45 +08:00
pooling.py	[autoparallel]integrate auto parallel feature with new tracer (#3408)	2023-04-04 17:40:45 +08:00
tensor.py	[autoparallel]integrate auto parallel feature with new tracer (#3408)	2023-04-04 17:40:45 +08:00
where.py	[autoparallel]integrate auto parallel feature with new tracer (#3408)	2023-04-04 17:40:45 +08:00