File tree

• .github
  • scripts
    • generate_binary_build_matrix.py
  • workflows
    • generated-linux-binary-conda-nightly.yml
    • generated-linux-binary-manywheel-master.yml
    • generated-windows-binary-wheel-nightly.yml
    • inductor-perf-smoke-test.yml
    • inductor-perf-test-nightly.yml
    • inductor.yml
    • periodic.yml
    • pull.yml
    • trunk.yml
• benchmarks/dynamo
  • common.py
  • torchbench.py
• test
  • distributed
    • _tensor
      • test_dtensor_ops.py
    • fsdp
      • test_fsdp_state_dict.py
      • test_fsdp_summon_full_params.py
      • test_fsdp_unshard_params.py
    • test_dynamo_distributed.py
  • dynamo
    • test_minifier.py
  • inductor
    • test_perf.py
    • test_torchinductor.py
• torch
  • _dynamo
    • __init__.py
    • backends
      • __init__.py
      • distributed.py
      • registry.py
    • convert_frame.py
    • debug_utils.py
    • eval_frame.py
    • guards.py
    • optimizations
      • __init__.py
      • backends.py
      • training.py
    • output_graph.py
    • testing.py
    • utils.py
    • variables
      • builder.py
  • _functorch
    • aot_autograd.py
  • _inductor
    • codegen
      • wrapper.py
    • graph.py
    • ir.py
    • lowering.py
    • scheduler.py
    • sizevars.py
  • distributed/fsdp
    • _common_utils.py
    • _optim_utils.py
    • _runtime_utils.py
    • _state_dict_utils.py
    • _traversal_utils.py
    • _unshard_param_utils.py
    • fully_sharded_data_parallel.py
  • fx/passes
    • shape_prop.py
  • testing/_internal
    • common_fsdp.py