mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
47cdad2995
Enabling 26 UTs for ROCm in the following files: - distributed._shard.sharded_optim.test_sharded_optim - 2 UTs - distributed._shard.sharded_tensor.ops.test_binary_cmp - 4 UTs - distributed._shard.sharded_tensor.ops.test_init - 3 UTs - distributed._shard.sharded_tensor.ops.test_embedding - 2 UTs - distributed._shard.sharded_tensor.ops.test_embedding_bag - 2 UTs - distributed._composable.test_replicate_with_compiler - 4 UTs - distributed._composable.fsdp.test_fully_shard_grad_scaler - 1 UTs - distributed.tensor.test_attention - 4 UTs - distributed.tensor.test_matrix_ops - 1 UTs - distributed.tensor.test_tensor_ops - 1 UTs - distributed.fsdp.test_fsdp_grad_acc - 2 UTs Pull Request resolved: https://github.com/pytorch/pytorch/pull/149369 Approved by: https://github.com/jeffdaily |
||
|---|---|---|
| .. | ||
| test_checkpoint_wrapper.py | ||
| test_distributed_checkpoint.py | ||
| test_fsdp_apply.py | ||
| test_fsdp_backward_prefetch.py | ||
| test_fsdp_checkpoint.py | ||
| test_fsdp_clip_grad_norm.py | ||
| test_fsdp_comm_hooks.py | ||
| test_fsdp_comm.py | ||
| test_fsdp_core.py | ||
| test_fsdp_dtensor_state_dict.py | ||
| test_fsdp_exec_order.py | ||
| test_fsdp_fine_tune.py | ||
| test_fsdp_flatten_params.py | ||
| test_fsdp_freezing_weights.py | ||
| test_fsdp_fx.py | ||
| test_fsdp_grad_acc.py | ||
| test_fsdp_hybrid_shard.py | ||
| test_fsdp_ignored_modules.py | ||
| test_fsdp_input.py | ||
| test_fsdp_memory.py | ||
| test_fsdp_meta.py | ||
| test_fsdp_misc.py | ||
| test_fsdp_mixed_precision.py | ||
| test_fsdp_multiple_forward.py | ||
| test_fsdp_multiple_wrapping.py | ||
| test_fsdp_optim_state.py | ||
| test_fsdp_overlap.py | ||
| test_fsdp_pure_fp16.py | ||
| test_fsdp_sharded_grad_scaler.py | ||
| test_fsdp_state_dict.py | ||
| test_fsdp_tp_integration.py | ||
| test_fsdp_traversal.py | ||
| test_fsdp_uneven.py | ||
| test_fsdp_unshard_params.py | ||
| test_fsdp_use_orig_params.py | ||
| test_hsdp_dtensor_state_dict.py | ||
| test_shard_utils.py | ||
| test_utils.py | ||
| test_wrap.py | ||