mirror of
https://github.com/zebrajr/pytorch.git
synced 2026-01-15 12:15:51 +00:00
bdad80fbe3
This reverts commit 8982850fb6.
Reverted https://github.com/pytorch/pytorch/pull/168157 on behalf of https://github.com/izaitsevfb due to needs to be evaluated for NE regression internally ([comment](https://github.com/pytorch/pytorch/pull/168157#issuecomment-3648542973))
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()
|