mirror of
https://github.com/zebrajr/pytorch.git
synced 2026-01-15 12:15:51 +00:00
[ROCm] skip AsyncTP test class as AsyncTP is not supported on ROCm (#166316)
Fixes #ISSUE_NUMBER Pull Request resolved: https://github.com/pytorch/pytorch/pull/166316 Approved by: https://github.com/jeffdaily
This commit is contained in:
committed by
PyTorch MergeBot
parent
0eacd934bc
commit
ac841267a1
@@ -279,6 +279,7 @@ class SymmetricMemoryTest(MultiProcContinuousTest):
|
||||
# MultiProcContinuousTest will skip all the following tests if a test fails (
|
||||
# we should fix this too). We still want to get the test signals for the core
|
||||
# symmetric memory APIs when Async TP ops fail.
|
||||
@skip_if_rocm_multiprocess # AsyncTP is not yet supported on ROCm
|
||||
@instantiate_parametrized_tests
|
||||
@requires_cuda_p2p_access()
|
||||
class AsyncTPTest(MultiProcContinuousTest):
|
||||
|
||||
Reference in New Issue
Block a user