mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 12:54:11 +08:00
afe15d2d2f
See #127836 for details. Pull Request resolved: https://github.com/pytorch/pytorch/pull/127840 Approved by: https://github.com/oulgen
15 lines
348 B
Python
15 lines
348 B
Python
# mypy: allow-untyped-defs
|
|
import torch._C._lazy
|
|
|
|
|
|
def dump(dot_file_name: str):
|
|
"""Dump TrieCache in the dot format"""
|
|
return torch._C._lazy._dump_ir_cache(dot_file_name)
|
|
|
|
|
|
def reset():
|
|
"""Clear TrieCache. This is needed in testing to avoid
|
|
node reusing between different tests.
|
|
"""
|
|
return torch._C._lazy._clear_ir_cache()
|