mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
12cb26509a
This is a big bang PR, merge conflicts are probably expected and will be addressed at merge. Pull Request resolved: https://github.com/pytorch/pytorch/pull/81643 Approved by: https://github.com/ezyang
14 lines
321 B
Python
14 lines
321 B
Python
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()
|