Home
last modified time | relevance | path

Searched refs:exception_msg_pattern (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/test/lazy/
H A Dtest_extract_compiled_graph.py132 def verify_reusing_compiled_graph(mod, exception_msg_pattern, ncase=10): argument
141 if exception_msg_pattern is None:
144 if not re.search(exception_msg_pattern, exception_message):
152 if exception_msg_pattern is not None:
184 def maketest(module_cls, exception_msg_pattern=None, ctxmgr=None): argument
190 verify_reusing_compiled_graph(module_cls(), exception_msg_pattern)
201 exception_msg_pattern="fallback.*aten::sub",