File tree

• benchmarks/distributed/pipeline
  • benchmark_dataset.py
  • pipe.py
• docs/source
  • conf.py
  • distributed.rst
  • pipeline.rst
• test
  • allowlist_for_publicAPI.json
  • distributed/pipeline/sync
    • LICENSE
    • __init__.py
    • conftest.py
    • skip
      • __init__.py
      • test_api.py
      • test_gpipe.py
      • test_inspect_skip_layout.py
      • test_leak.py
      • test_portal.py
      • test_stash_pop.py
      • test_tracker.py
      • test_verify_skippables.py
    • test_balance.py
    • test_bugs.py
    • test_checkpoint.py
    • test_copy.py
    • test_deferred_batch_norm.py
    • test_dependency.py
    • test_inplace.py
    • test_microbatch.py
    • test_phony.py
    • test_pipe.py
    • test_pipeline.py
    • test_stream.py
    • test_transparency.py
    • test_worker.py
  • test_public_bindings.py
  • test_testing.py
• torch
  • distributed/pipeline
    • __init__.py
    • sync
      • LICENSE
      • __init__.py
      • _balance
        • __init__.py
        • blockpartition.py
        • profile.py
        • py.typed
      • batchnorm.py
      • checkpoint.py
      • copy.py
      • dependency.py
      • microbatch.py
      • phony.py
      • pipe.py
      • pipeline.py
      • py.typed
      • skip
        • __init__.py
        • layout.py
        • namespace.py
        • portal.py
        • skippable.py
        • tracker.py
      • stream.py
      • utils.py
      • worker.py
  • testing/_internal/distributed
    • pipe_with_ddp_test.py