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()
|