ColossalAI/op_builder
Frank Lee 8711310cda
[setup] remove torch dependency (#2333)
2023-01-05 13:53:28 +08:00
..
__init__.py
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