13 lines
275 B
Python
13 lines
275 B
Python
|
import cacheflow.parallel_utils.parallel_state
|
||
|
|
import cacheflow.parallel_utils.tensor_parallel
|
||
|
|
import cacheflow.parallel_utils.utils
|
||
|
|
|
||
|
|
# Alias parallel_state as mpu, its legacy name
|
||
|
|
mpu = parallel_state
|
||
|
|
|
||
|
|
__all__ = [
|
||
|
|
"parallel_state",
|
||
|
|
"tensor_parallel",
|
||
|
|
"utils",
|
||
|
|
]
|