package info (click to toggle)
pytorch-cuda 2.6.0%2Bdfsg-7
  • links: PTS, VCS
  • area: contrib
  • in suites: forky, sid, trixie
  • size: 161,620 kB
  • sloc: python: 1,278,832; cpp: 900,322; ansic: 82,710; asm: 7,754; java: 3,363; sh: 2,811; javascript: 2,443; makefile: 597; ruby: 195; xml: 84; objc: 68

Folder: fsdp

d .. (parent)
d d rwxr-xr-x 4,096 _fully_shard
- - rw-r--r-- 1,740 __init__.py
- - rw-r--r-- 22,461 _common_utils.py
- - rw-r--r-- 5,706 _debug_utils.py
- - rw-r--r-- 2,674 _dynamo_utils.py
- - rw-r--r-- 16,122 _exec_order_utils.py
- - rw-r--r-- 123,608 _flat_param.py
- - rw-r--r-- 4,969 _fsdp_extensions.py
- - rw-r--r-- 46,274 _init_utils.py
- - rw-r--r-- 1,081 _limiter_utils.py
- - rw-r--r-- 86,934 _optim_utils.py
- - rw-r--r-- 66,196 _runtime_utils.py
- - rw-r--r-- 4,623 _shard_utils.py
- - rw-r--r-- 34,366 _state_dict_utils.py
- - rw-r--r-- 10,776 _trace_utils.py
- - rw-r--r-- 4,641 _traversal_utils.py
- - rw-r--r-- 11,497 _unshard_param_utils.py
- - rw-r--r-- 10,925 _wrap_utils.py
- - rw-r--r-- 18,880 api.py
- - rw-r--r-- 100,299 fully_sharded_data_parallel.py
- - rw-r--r-- 16,148 sharded_grad_scaler.py
- - rw-r--r-- 22,643 wrap.py