alpa
alpa copied to clipboard
[FEATURE] launch physical meshes after compilation
This can be a starting point to learn runtime_emitter
.
Background
In Pipeshard Parallel, the final compilation step is to interpret the solution into a configuration containing all information about computation, communication, parameter, output and debug information. Now the instruction_lists
and executable_configs
are indexed by MeshHostWorker
, which means we have to give the emitter a physical mesh group(see here). The final target is to run all steps in the compilation(except for profiling) without using GPU resources.
TODO
- [ ] call
SymbolicReshardingTask.put_tasks
inPipeshardMeshDriverExecutable.__init__
. The same applies for theSymbolicBroadcastReshardingTask
; - [ ] replace the
mesh_group
bysliced_virtual_meshes
and move the two lines out of the function. Maybe we can create aVirtualMeshGroup
to provide interfaces. Some operations will be skipped; - [ ] add skipped operations from last step in
PipeshardMeshDriverExecutable.__init__
.
working on it