MCPcopy Index your code
hub / github.com/pytorch/examples / fsdp_tp_example.py

File fsdp_tp_example.py

distributed/tensor_parallelism/fsdp_tp_example.py:None–None  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

1"""
2This is the script to test 2D Parallel which combines Tensor/Sequence
3parallel with Fully Sharded Data Parallel (TP/SP + FSDP) on a example
4Llama2 model. We show an E2E working flow from forward, backward

Callers

nothing calls this directly

Calls 6

verify_min_gpu_countFunction · 0.90
get_loggerFunction · 0.90
rank_logFunction · 0.90
ModelArgsClass · 0.90
from_model_argsMethod · 0.80
init_weightsMethod · 0.45

Tested by

no test coverage detected