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.
ColossalAI/op_builder
Frank Lee 9b765e7a69
[setup] removed the build dependency on colossalai (#2307)
2 years ago
..
__init__.py [setup] removed the build dependency on colossalai (#2307) 2 years ago
builder.py [setup] removed the build dependency on colossalai (#2307) 2 years ago
cpu_adam.py [setup] removed the build dependency on colossalai (#2307) 2 years ago
fused_optim.py [setup] removed the build dependency on colossalai (#2307) 2 years ago
moe.py [setup] removed the build dependency on colossalai (#2307) 2 years ago
multi_head_attn.py [setup] removed the build dependency on colossalai (#2307) 2 years ago
scaled_upper_triang_masked_softmax.py [setup] removed the build dependency on colossalai (#2307) 2 years ago
utils.py [setup] removed the build dependency on colossalai (#2307) 2 years ago