mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
846963fa9b
This reverts commit b9d7de3a094598c3dc0dd52e57bce30eb684c9d8.
Reverted https://github.com/pytorch/pytorch/pull/158137 on behalf of https://github.com/malfet due to Broke inductor torchbench, see 663da17b62/1 ([comment](https://github.com/pytorch/pytorch/pull/158137#issuecomment-3191841298))
20 lines
603 B
Python
20 lines
603 B
Python
#!/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()
|