Frank Lee 8711310cda [setup] remove torch dependency (#2333)		2023-01-05 13:53:28 +08:00
..
__init__.py	[setup] removed the build dependency on colossalai (#2307)	2023-01-04 11:38:42 +08:00
builder.py	[setup] remove torch dependency (#2333)	2023-01-05 13:53:28 +08:00
cpu_adam.py	[builder] reconfig op_builder for pypi install (#2314)	2023-01-04 16:32:32 +08:00
fused_optim.py	[builder] reconfig op_builder for pypi install (#2314)	2023-01-04 16:32:32 +08:00
moe.py	[builder] reconfig op_builder for pypi install (#2314)	2023-01-04 16:32:32 +08:00
multi_head_attn.py	[builder] reconfig op_builder for pypi install (#2314)	2023-01-04 16:32:32 +08:00
scaled_upper_triang_masked_softmax.py	[builder] reconfig op_builder for pypi install (#2314)	2023-01-04 16:32:32 +08:00
utils.py	[setup] removed the build dependency on colossalai (#2307)	2023-01-04 11:38:42 +08:00