1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
|
# 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()
|
