Workloads Documentation#
This section contains automatically generated documentation for all CloudAI workloads. Each workload provides specific functionality for running different types of tests and benchmarks.
Available Workloads#
Test |
Slurm |
Kubernetes |
RunAI |
Standalone |
|---|---|---|---|---|
❌ |
❌ |
❌ |
✅ |
|
✅ |
✅ |
❌ |
❌ |
|
✅ |
❌ |
❌ |
❌ |
|
✅ |
❌ |
❌ |
❌ |
|
✅ |
❌ |
❌ |
❌ |
|
✅ |
❌ |
❌ |
❌ |
|
✅ |
❌ |
❌ |
❌ |
|
MegatronRun |
✅ |
❌ |
❌ |
❌ |
✅ |
❌ |
❌ |
❌ |
|
✅ |
✅ |
✅ |
❌ |
|
✅ |
❌ |
❌ |
❌ |
|
✅ |
❌ |
❌ |
❌ |
|
✅ |
❌ |
❌ |
❌ |
|
✅ |
❌ |
❌ |
❌ |
|
✅ |
❌ |
❌ |
❌ |
|
✅ |
✅ |
❌ |
✅ |
|
✅ |
❌ |
❌ |
❌ |
|
Triton Inference |
✅ |
❌ |
❌ |
❌ |
✅ |
❌ |
❌ |
❌ |
Adding New Workloads#
To add documentation for a new workload:
Add docstrings to your Python classes and methods.
Create a reStructuredText file in
doc/workloads/(e.g.,my_workload.rst).Add it to the table above.
The documentation will be automatically generated during the build process.