mirror of
https://github.com/microsoft/mscclpp.git
synced 2026-05-11 17:00:22 +00:00
be6a941fba
The PR contains following changes: Python side: - Channel based DSL implementation: decouple channel with chunk. - Users create channel explicitly, only need local_rank, remote_rank and channel_type - Adjust executor json file, add remote_buffer fields, different op can use different channel and remote buffers combination. - Reimplement operation fusion, data dependency check mechanism - Add new op such as semaphore, pipeline - Clean code and enhance document C++ side: - Support new execution file json format - Support semaphore and pipeline operation - code clean, support non-zero copy scenario --------- Co-authored-by: Caio Rocha <caiorocha@microsoft.com> Co-authored-by: Changho Hwang <changhohwang@microsoft.com>
1.1 KiB
1.1 KiB
applyTo
| applyTo |
|---|
| python/mscclpp/language/*.py |
Instructions for DSL API Documentation
Overview
The MSCCL++ DSL (Domain Specific Language) provides a Python API for defining distributed GPU communication patterns. All API functions should have comprehensive Google-style docstrings.
Documentation Requirements
- Add google-style docstrings to the DSL API functions in the
mscclpp.languagepackage. - Ensure that each function's docstring includes:
- A brief description of what the function does.
- Parameters with their types and descriptions.
- Return type and description.
- Any exceptions raised by the function, if applicable.
- Usage examples where appropriate.
Implementation Steps
- Open each Python file in the
python.mscclpp.languagefolder, exclude__init__.pyand internal folders. - For each function in the file, add a Google-style docstring that follows the documentation requirements outlined above.
- Ensure that the docstrings are clear, concise, and accurately describe the function's behavior.
- Review the docstrings for consistency in style and formatting.