mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 12:54:11 +08:00
b38be727eb
[Inductor UT] Reuse Inductor test case for Intel GPU. Reuse `test/inductor/test_torchinductor_opinfo.py` Reuse `test/inductor/test_minifier_isolate.py` Pull Request resolved: https://github.com/pytorch/pytorch/pull/134556 Approved by: https://github.com/etaf, https://github.com/eellison
60 lines
2.0 KiB
Python
60 lines
2.0 KiB
Python
# Owner(s): ["module: inductor"]
|
|
import unittest
|
|
|
|
import torch._inductor.config as inductor_config
|
|
from torch._dynamo.test_minifier_common import MinifierTestBase
|
|
from torch.testing._internal.common_utils import (
|
|
IS_JETSON,
|
|
IS_MACOS,
|
|
skipIfRocm,
|
|
skipIfWindows,
|
|
skipIfXpu,
|
|
TEST_WITH_ASAN,
|
|
)
|
|
from torch.testing._internal.inductor_utils import GPU_TYPE
|
|
from torch.testing._internal.triton_utils import requires_gpu
|
|
|
|
|
|
# These minifier tests are slow, because they must be run in separate
|
|
# subprocesses
|
|
class MinifierIsolateTests(MinifierTestBase):
|
|
def _test_after_aot_runtime_error(self, device, expected_error):
|
|
run_code = f"""\
|
|
@torch.compile()
|
|
def inner(x):
|
|
x = torch.relu(x)
|
|
x = torch.cos(x)
|
|
return x
|
|
|
|
inner(torch.randn(2, 2).to("{device}"))
|
|
"""
|
|
# These must isolate because they crash the process
|
|
self._run_full_test(run_code, "aot", expected_error, isolate=True)
|
|
|
|
@unittest.skipIf(IS_JETSON, "Fails on Jetson")
|
|
@inductor_config.patch("cpp.inject_relu_bug_TESTING_ONLY", "runtime_error")
|
|
@skipIfWindows(
|
|
msg="Build Failed: fatal error C1083: Cannot open include file: 'Python.h': No such file or directory"
|
|
)
|
|
def test_after_aot_cpu_runtime_error(self):
|
|
self._test_after_aot_runtime_error("cpu", "")
|
|
|
|
@skipIfRocm
|
|
@skipIfXpu
|
|
@requires_gpu
|
|
@inductor_config.patch("triton.inject_relu_bug_TESTING_ONLY", "runtime_error")
|
|
def test_after_aot_gpu_runtime_error(self):
|
|
self._test_after_aot_runtime_error(GPU_TYPE, "device-side assert")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
import sys
|
|
|
|
from torch._dynamo.test_case import run_tests
|
|
|
|
# Skip CI tests on mac since CPU inductor does not seem to work due to C++ compile errors,
|
|
# also skip on ASAN due to https://github.com/pytorch/pytorch/issues/98262
|
|
# also skip on Py 3.11+ since unhandled exceptions can cause segfaults
|
|
if not IS_MACOS and not TEST_WITH_ASAN and sys.version_info < (3, 11):
|
|
run_tests()
|