Movatterモバイル変換

[0]ホーム
URL:

Skip to content

Navigation Menu

Sign in
Appearance settings

Search code, repositories, users, issues, pull requests...

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly

 Sign up
Appearance settings

Commite0b8325

Browse files
jeffdailypytorchmergebot
authored andcommitted
[ROCm][CI] additional PLATFORM_SUPPORTS_SYMM_MEM skips (#170630)
test/distributed/test_symmetric_memory.py added tests but did not add skips for unsupported platforms. Specifically affects ROCm.Pull Requestresolved:#170630Approved by:https://github.com/jeffdailyCo-authored-by: Jeff Daily <jeff.daily@amd.com>
1 parent5424f6c commite0b8325

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

‎test/distributed/test_symmetric_memory.py‎

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1376,6 +1376,9 @@ def _init_process(self):
13761376
torch.cuda.set_device(self.device)
13771377
torch.manual_seed(42+self.rank)
13781378

1379+
@skipIf(
1380+
notPLATFORM_SUPPORTS_SYMM_MEM,"SymmMem is not supported on this ROCm arch"
1381+
)
13791382
@skip_if_lt_x_gpu(2)
13801383
deftest_mempool_tensor_factory(self):
13811384
self._init_process()
@@ -1398,6 +1401,9 @@ def test_mempool_tensor_factory(self):
13981401
)
13991402
self.assertEqual(tensor,expected)
14001403

1404+
@skipIf(
1405+
notPLATFORM_SUPPORTS_SYMM_MEM,"SymmMem is not supported on this ROCm arch"
1406+
)
14011407
@skip_if_lt_x_gpu(2)
14021408
deftest_mempool_compute_ops(self):
14031409
self._init_process()

0 commit comments

Comments
 (0)
[8]ページ先頭
©2009-2025 Movatter.jp