Xilun Wu c4d835fbab [DTensor][conv] add DTensor convolution_backward op support for case where the input Tensor has requires_grad=False (#142278) ... Fixes #142058 ## Summary DTensor `convolution_backward` op throws exception when the input Tensor has `requires_grad=False` which happens if the conv layer is the first layer in the model. ATEN convolution_backward op Usually returns 3 Tensors (grad_input, grad_weight, grad_bias) and the `grad_input` is actually an Optional[Tensor] which can be `None` in the case mentioned above. However, the DTensor sharding propagation rule and corresponding TP conv backward implementation both assume that the `grad_input` would be existent. ## Fix allow the `grad_input` to be `None` for `convolution_backward` op. ## Test `pytest test/distributed/tensor/test_convolution_ops.py` ## Follow-up The current implementation of DTensor conv op also ignores `output_mask` and this may need further care. Pull Request resolved: https://github.com/pytorch/pytorch/pull/142278 Approved by: https://github.com/bdhirsh		2025-02-10 07:06:40 +00:00
..
_composable	Refactoring Distributed test cases to be device agnostic [1/n] (#145222)	2025-02-05 18:47:09 +00:00
_shard
_tools
algorithms	Enable ruff F841 on distributed tests (#146131)	2025-02-01 03:06:16 +00:00
bin
checkpoint	[DCP] Remove all-gather of state dict keys (#145998)	2025-02-04 03:16:13 +00:00
elastic	Enable ruff F841 on distributed tests (#146131)	2025-02-01 03:06:16 +00:00
flight_recorder
fsdp	[BE][Ez]: ISC001 Auto concatenate implicit one line strings (#146408)	2025-02-04 19:07:04 +00:00
launcher
nn/jit
optim
pipelining	Remove stage_index_to_group_rank from schedule (#146217)	2025-02-05 21:26:45 +00:00
rpc
tensor	[DTensor][conv] add DTensor convolution_backward op support for case where the input Tensor has requires_grad=False (#142278)	2025-02-10 07:06:40 +00:00
argparse_util_test.py
test_backends.py
test_c10d_common.py
test_c10d_functional_native.py	PyWork: preserve Python reference counting when used in functional collectives (#146376)	2025-02-07 18:07:53 +00:00
test_c10d_gloo.py
test_c10d_logger.py
test_c10d_nccl.py	[DDP] Use NCCL allocated memory for gradient bucket (#146589)	2025-02-10 05:23:11 +00:00
test_c10d_object_collectives.py
test_c10d_ops_nccl.py
test_c10d_pypg.py
test_c10d_spawn.py
test_c10d_spawn_gloo.py
test_c10d_spawn_nccl.py
test_c10d_spawn_ucc.py
test_c10d_ucc.py
test_collective_utils.py
test_composability.py
test_compute_comm_reordering.py
test_control_collectives.py
test_data_parallel.py
test_device_mesh.py
test_distributed_spawn.py
test_dynamo_distributed.py	dynamo: fsdp throw unimplemented vs attribute error (#146188)	2025-02-04 21:45:55 +00:00
test_fake_pg.py
test_functional_api.py
test_inductor_collectives.py
test_launcher.py
test_multi_threaded_pg.py
test_nccl.py
test_pg_wrapper.py
test_serialization.py	distributed/serialization: add experimental streaming torch.save/load methods (#146555)	2025-02-07 18:08:11 +00:00
test_store.py
test_symmetric_memory.py	[SymmetricMemory] fix an issue where rendezvous is performed with wrong device context when torch.cuda.set_device() is not callled (#144886)	2025-01-28 01:43:37 +00:00