mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +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
22 lines
711 B
Python
22 lines
711 B
Python
import torch._C._lazy
|
|
|
|
|
|
def render_ir_graph(tensors):
|
|
"""Return a text dump of the LTC IR graph in dot format for the tensors.
|
|
The text can be processed by tools like dot to be rendered in pdf,png etc."""
|
|
return torch._C._lazy._get_tensors_dot(tensors)
|
|
|
|
|
|
def dump_ir(tensors, ir_format):
|
|
"""Return a dump of the tensors in the specified format.
|
|
Valid format are
|
|
- text: for LTC IR
|
|
- backend: for the activate backend IR
|
|
"""
|
|
if ir_format == "text":
|
|
return torch._C._lazy._get_tensors_text(tensors)
|
|
elif ir_format == "backend":
|
|
return torch._C._lazy._get_tensors_backend(tensors)
|
|
else:
|
|
raise RuntimeError(f"Unrecognized IR format: {ir_format}")
|