fhelium.benchmarks.standalone.distributed_worker
Internal torchrun worker for distributed benchmarks.
COLLECTIVES_WORKLOAD_ID
constant View source
python
COLLECTIVES_WORKLOAD_ID = 'spmd-collectives'1
ROTATION_MATVEC_WORKLOAD_ID
constant View source
python
ROTATION_MATVEC_WORKLOAD_ID = 'spmd-ckks-rotation-workload'1
main
function View source
python
def main() -> None: ...1