You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

YeAnbang 3883db452c [NFC] polish unary_elementwise_generator.py code style (#4267) ... Co-authored-by: aye42 <aye42@gatech.edu>		1 year ago
..
__init__.py	[autoparallel] refactor handlers which reshape input tensors (#2615)	2 years ago
batch_norm_generator.py	fix typo colossalai/auto_parallel autochunk fx/passes etc. (#3808)	2 years ago
binary_elementwise_generator.py	[NFC] fix typo with colossalai/auto_parallel/tensor_shard (#3742)	2 years ago
conv_strategy_generator.py	fix typo colossalai/auto_parallel autochunk fx/passes etc. (#3808)	2 years ago
embedding_generator.py	[autoparallel]add embedding handler (#2089)	2 years ago
getattr_generator.py	[autoparallel] update_getattr_handler (#2193)	2 years ago
getitem_generator.py	[autoparallel] update getitem handler (#2207)	2 years ago
layer_norm_generator.py	fix typo colossalai/auto_parallel autochunk fx/passes etc. (#3808)	2 years ago
matmul_strategy_generator.py	[test] fixed tests failed due to dtensor change (#4082)	1 year ago
normal_pooling_generator.py	fix typo colossalai/auto_parallel autochunk fx/passes etc. (#3808)	2 years ago
output_generator.py	[autoparallel] add split handler (#2032)	2 years ago
placeholder_generator.py	…
reshape_generator.py	[hotfix] add shard dim to aviod backward communication error (#2954)	2 years ago
softmax_generator.py	[autoparallel] implement softmax handler (#2132)	2 years ago
strategy_generator.py	[NFC] fix typo with colossalai/auto_parallel/tensor_shard (#3742)	2 years ago
sum_generator.py	[autoparallel] add sum handler (#2101)	2 years ago
tensor_constructor_generator.py	[autoparallel] add tensor constructor handler (#2082)	2 years ago
unary_elementwise_generator.py	[NFC] polish unary_elementwise_generator.py code style (#4267)	1 year ago
where_generator.py	…