You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
15 lines
348 B
15 lines
348 B
# 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()
|