frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Skip tests on ROCm (#53339)

Browse Source 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/53339

Skip tests on ROCm

Test Plan: CI

Reviewed By: gdankel, ZolotukhinM

Differential Revision: D26838813

fbshipit-source-id: e26286a61a192710e393c19d3eb2316b6c76a42e
This commit is contained in:
Ilia Cherniavskii
2021-03-04 21:53:27 -08:00
committed by Facebook GitHub Bot
parent 8bac382d9d
commit dfd5331e9c
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
test/test_profiler.py
View File

@ -10,7 +10,8 @@ import torch.optim
import torch.utils.data
from torch.testing._internal.common_cuda import TEST_MULTIGPU
from torch.testing._internal.common_utils import (
TestCase, run_tests, TEST_WITH_ASAN, IS_WINDOWS, TemporaryFileName, TemporaryDirectoryName)
TestCase, run_tests, TEST_WITH_ASAN, TEST_WITH_ROCM, IS_WINDOWS,
TemporaryFileName, TemporaryDirectoryName)
from torch.autograd.profiler import profile as _profile
from torch.profiler import (
kineto_available, profile, record_function, DeviceType, ProfilerActivity
@ -118,7 +119,7 @@ class TestProfiler(TestCase):
@unittest.skipIf(not kineto_available(), "Kineto is required")
def test_kineto(self):
use_cuda = torch.cuda.is_available()
use_cuda = torch.cuda.is_available() and (not TEST_WITH_ROCM)
with _profile(use_cuda=use_cuda, use_kineto=True):
self.payload(use_cuda=use_cuda)
@ -147,6 +148,7 @@ class TestProfiler(TestCase):
@unittest.skipIf(not kineto_available(), "Kineto is required")
@unittest.skipIf(not TEST_MULTIGPU, "Multiple GPUs needed")
@unittest.skipIf(TEST_WITH_ROCM, "Not supported on ROCm")
def test_kineto_multigpu(self):
with profile(
activities=[