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 8711310cda
[setup] remove torch dependency (#2333)
2 years ago
..
__init__.py [setup] removed the build dependency on colossalai (#2307) 2 years ago
builder.py [setup] remove torch dependency (#2333) 2 years ago
cpu_adam.py [builder] reconfig op_builder for pypi install (#2314) 2 years ago
fused_optim.py [builder] reconfig op_builder for pypi install (#2314) 2 years ago
moe.py [builder] reconfig op_builder for pypi install (#2314) 2 years ago
multi_head_attn.py [builder] reconfig op_builder for pypi install (#2314) 2 years ago
scaled_upper_triang_masked_softmax.py [builder] reconfig op_builder for pypi install (#2314) 2 years ago
utils.py [setup] removed the build dependency on colossalai (#2307) 2 years ago