mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
93a33bf3ac
Changes: 1. Make some arguments positional-only as we only support Python 3.8+ 2. Clean up `torch.typename(obj)` implementation. 3. Update type annotations., especially `is_tensor()` and `is_masked_tensor()` using `TypeGuard`. Pull Request resolved: https://github.com/pytorch/pytorch/pull/129001 Approved by: https://github.com/malfet
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()
|