mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
[ROCm] Enabling several UTs (#161715)
All these UTs are working as is, just removing the skip - test_p2p_ipc - test_repros.py: working, added fp8 support - test_activation_checkpointing.py - test_content_store.py - test_cuda_multigpu.py - test_compute_comm_reordering.py - test_segment_reductions.py - test_dataloader.py - test_math_ops.py - test_loop_ordering.py - test_control_flow.py - distributed_test.py - test_mem_tracker.py - test_fsdp_optim_state.py - test_fully_shard_mixed_precision.py: skippped for < ROCm7.0 - test_aot_inductor_custom_ops.py - test_c10d_ops_nccl.py - test_eager_transforms.py - test_sparse_csr.py - test_inductor_collectives.py - test_fake_tensor.py - test_cupy_as_tensor.py - test_cuda.py: enable UTs that are working - test_matmul_cuda.py: enable UTs that are working Fixes #ISSUE_NUMBER Pull Request resolved: https://github.com/pytorch/pytorch/pull/161715 Approved by: https://github.com/msaroufim Co-authored-by: Mark Saroufim <marksaroufim@fb.com>
This commit is contained in:
committed by
PyTorch MergeBot
parent
3ea6868049
commit
c0142f5c06
@ -14,7 +14,6 @@ from torch.testing._internal.common_utils import (
|
||||
run_tests,
|
||||
gradcheck,
|
||||
parametrize,
|
||||
skipIfRocm,
|
||||
)
|
||||
|
||||
|
||||
@ -231,7 +230,6 @@ class TestSegmentReductions(TestCase):
|
||||
length_type,
|
||||
)
|
||||
|
||||
@skipIfRocm
|
||||
@dtypes(
|
||||
*product(
|
||||
(torch.half, torch.bfloat16, torch.float, torch.double),
|
||||
|
Reference in New Issue
Block a user