1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
|
#!/usr/bin/env python3
import os
from torch._inductor import pattern_matcher
from torch._inductor.fx_passes import joint_graph
if __name__ == "__main__":
# Start by deleting all the existing patterns.
for path in pattern_matcher.SERIALIZED_PATTERN_PATH.iterdir():
if path.name in {"__init__.py", "__pycache__"}:
continue
if path.is_file():
path.unlink()
# Now have joint_graph load all known patterns and tell the pattern matcher
# to serialize the patterns as it goes.
os.environ["PYTORCH_GEN_PATTERNS"] = "1"
joint_graph.lazy_init()
|
