mirror of https://github.com/hpcaitech/ColossalAI
8 lines
387 B
Python
8 lines
387 B
Python
|
from .strategy_generator import StrategyGenerator_V2
|
||
|
|
from .matmul_strategy_generator import DotProductStrategyGenerator, MatVecStrategyGenerator, LinearProjectionStrategyGenerator, BatchedMatMulStrategyGenerator
|
||
|
|
|
||
|
|
__all__ = [
|
||
|
|
'StrategyGenerator_V2', 'DotProductStrategyGenerator', 'MatVecStrategyGenerator',
|
||
|
|
'LinearProjectionStrategyGenerator', 'BatchedMatMulStrategyGenerator'
|
||
|
|
]
|