ColossalAI/op_builder
Frank Lee 9b765e7a69
[setup] removed the build dependency on colossalai (#2307)
2023-01-04 11:38:42 +08:00
..
__init__.py [setup] removed the build dependency on colossalai (#2307) 2023-01-04 11:38:42 +08:00
builder.py [setup] removed the build dependency on colossalai (#2307) 2023-01-04 11:38:42 +08:00
cpu_adam.py [setup] removed the build dependency on colossalai (#2307) 2023-01-04 11:38:42 +08:00
fused_optim.py [setup] removed the build dependency on colossalai (#2307) 2023-01-04 11:38:42 +08:00
moe.py [setup] removed the build dependency on colossalai (#2307) 2023-01-04 11:38:42 +08:00
multi_head_attn.py [setup] removed the build dependency on colossalai (#2307) 2023-01-04 11:38:42 +08:00
scaled_upper_triang_masked_softmax.py [setup] removed the build dependency on colossalai (#2307) 2023-01-04 11:38:42 +08:00
utils.py [setup] removed the build dependency on colossalai (#2307) 2023-01-04 11:38:42 +08:00