mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2026-05-03 01:19:15 +00:00
1cb532ffec
* [hotfix] correcnt cpu_optim runtime compilation * [builder] multihead attn * fix bug * fix a bug
6 lines
205 B
Python
6 lines
205 B
Python
from .cpu_adam import CPUAdamBuilder
|
|
from .fused_optim import FusedOptimBuilder
|
|
from .multi_head_attn import MultiHeadAttnBuilder
|
|
|
|
__all__ = ['CPUAdamBuilder', 'FusedOptimBuilder', 'MultiHeadAttnBuilder']
|