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