OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

[test] Disable TestXNNPACK on ROCM machines (#73476)

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

This is a temp fix, and needs further investigation.

Test Plan: CI

Reviewed By: kimishpatel

Differential Revision: D34501632

fbshipit-source-id: c90f51e6f27bce167c113e90694090728b561d74
(cherry picked from commit e6e62aafbe86bbd1bbc913af3725f8256a4b538e)
This commit is contained in:
Digant Desai 2022-02-26 16:15:14 -08:00 committed by PyTorch MergeBot
parent 5613527ef9
commit 09697c6df5
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
test/test_xnnpack_integration.py
View File

@ -14,12 +14,13 @@ from hypothesis import strategies as st
import io import io
import itertools import itertools
from torch.testing._internal.common_utils import TEST_WITH_TSAN from torch.testing._internal.common_utils import (TEST_WITH_TSAN, TEST_WITH_ROCM)
@unittest.skipUnless(torch.backends.xnnpack.enabled, @unittest.skipUnless(torch.backends.xnnpack.enabled,
" XNNPACK must be enabled for these tests." " XNNPACK must be enabled for these tests."
" Please build with USE_XNNPACK=1.") " Please build with USE_XNNPACK=1.")
@unittest.skipIf(TEST_WITH_TSAN, "TSAN fails with XNNPACK. Does not seem to have a good reason for failures.") @unittest.skipIf(TEST_WITH_TSAN, "TSAN fails with XNNPACK. Does not seem to have a good reason for failures.")
@unittest.skipIf(TEST_WITH_ROCM, "HACK, not sure why these fail on ROCM.")
class TestXNNPACKOps(TestCase): class TestXNNPACKOps(TestCase):
@given(batch_size=st.integers(0, 3), @given(batch_size=st.integers(0, 3),
data_shape=hu.array_shapes(1, 3, 2, 64), data_shape=hu.array_shapes(1, 3, 2, 64),
@ -181,6 +182,7 @@ class TestXNNPACKOps(TestCase):
" XNNPACK must be enabled for these tests." " XNNPACK must be enabled for these tests."
" Please build with USE_XNNPACK=1.") " Please build with USE_XNNPACK=1.")
@unittest.skipIf(TEST_WITH_TSAN, "TSAN fails with XNNPACK. Does not seem to have a good reason for failures.") @unittest.skipIf(TEST_WITH_TSAN, "TSAN fails with XNNPACK. Does not seem to have a good reason for failures.")
@unittest.skipIf(TEST_WITH_ROCM, "HACK, not sure why these fail on ROCM.")
class TestXNNPACKSerDes(TestCase): class TestXNNPACKSerDes(TestCase):
@given(batch_size=st.integers(0, 3), @given(batch_size=st.integers(0, 3),
data_shape=hu.array_shapes(1, 3, 2, 64), data_shape=hu.array_shapes(1, 3, 2, 64),
@ -572,6 +574,7 @@ class TestXNNPACKSerDes(TestCase):
" XNNPACK must be enabled for these tests." " XNNPACK must be enabled for these tests."
" Please build with USE_XNNPACK=1.") " Please build with USE_XNNPACK=1.")
@unittest.skipIf(TEST_WITH_TSAN, "TSAN fails with XNNPACK. Does not seem to have a good reason for failures.") @unittest.skipIf(TEST_WITH_TSAN, "TSAN fails with XNNPACK. Does not seem to have a good reason for failures.")
@unittest.skipIf(TEST_WITH_ROCM, "HACK, not sure why these fail on ROCM.")
class TestXNNPACKRewritePass(TestCase): class TestXNNPACKRewritePass(TestCase):
@staticmethod @staticmethod
def validate_transformed_module( def validate_transformed_module(
@ -933,6 +936,7 @@ class TestXNNPACKRewritePass(TestCase):
" XNNPACK must be enabled for these tests." " XNNPACK must be enabled for these tests."
" Please build with USE_XNNPACK=1.") " Please build with USE_XNNPACK=1.")
@unittest.skipIf(TEST_WITH_TSAN, "TSAN is not fork-safe since we're forking in a multi-threaded environment") @unittest.skipIf(TEST_WITH_TSAN, "TSAN is not fork-safe since we're forking in a multi-threaded environment")
@unittest.skipIf(TEST_WITH_ROCM, "HACK, not sure why these fail on ROCM.")
class TestXNNPACKConv1dTransformPass(TestCase): class TestXNNPACKConv1dTransformPass(TestCase):
@staticmethod @staticmethod
def validate_transform_conv1d_to_conv2d( def validate_transform_conv1d_to_conv2d(