pytorch/torch/distributed/_tensor/debug
2024-05-04 04:20:46 +00:00
..
__init__.py [dtensor] add CommDebugMode for debugging (#113592) 2023-11-27 02:40:28 +00:00
comm_mode.py [comm_mode] adding some initial c10d ops to CommDebugMode (#125475) 2024-05-04 04:20:46 +00:00
op_coverage.py Enable local_partial_types (#118467) 2024-01-28 13:38:22 +00:00
visualize_sharding.py [dtensor][debug] force visualize_sharding not to print for empty tensors (#121217) 2024-03-11 09:22:49 +00:00