feat(pytorch): instrument distributed training with pytorch.rank lifetime span#18449

Open

kr-igor wants to merge 16 commits into

mainfrom

kr-igor/pytorch-rank-span

Commits on Jun 10, 2026

feat(pytorch): instrument distributed training with pytorch.rank lifetime span
kr-igor
committed
feat(pytorch): wire optional step signals from Optimizer.step to C tracer
kr-igor
committed
chore: register DD_TRAINING_STEP_PROFILING in supported-configurations.json
kr-igor
committed
style: ruff format _distributed.py
kr-igor
committed
fix(test): use spec= in _make_fake_lib_no_steps to correctly raise AttributeError
kr-igor
committed
fix(test): rewrite step signal tests to inject module state directly
kr-igor
committed
feat(pytorch): propagate service to C tracer and rank span
kr-igor
committed
fix(pytorch): use int_service for pytorch.rank to correctly inherit DD_SERVICE
kr-igor
committed
fix(pytorch): propagate span.service to C tracer instead of ddtrace.config.service
kr-igor
committed
fix: remove unused ext_service import from _rank_root.py
kr-igor
committed

Commits on Jun 12, 2026

test(pytorch): verify no interference with torch.profiler.profile()
kr-igor
committed