feat: add option to print optimizer output

This commit is contained in:
Umut
2022-07-08 17:35:41 +02:00
parent fb9aa85aff
commit 83f18e262a
4 changed files with 70 additions and 15 deletions

View File

@@ -66,10 +66,10 @@ def test_compiler_verbose_compile(helpers, capsys):
return x + 42
inputset = range(10)
function.compile(inputset, configuration, artifacts, show_graph=True, show_mlir=True)
function.compile(inputset, configuration, artifacts, verbose=True)
captured = capsys.readouterr()
assert captured.out.strip() == (
assert captured.out.strip().startswith(
f"""
Computation Graph
@@ -80,7 +80,48 @@ Computation Graph
MLIR
--------------------------------------------------------------------------------
{artifacts.mlir_to_compile}
--------------------------------------------------------------------------------
Optimizer
--------------------------------------------------------------------------------
""".strip()
)
def test_compiler_verbose_virtual_compile(helpers, capsys):
"""
Test `compile` method of `compiler` decorator with verbose flag.
"""
configuration = helpers.configuration()
artifacts = DebugArtifacts()
@compiler({"x": "encrypted"})
def function(x):
return x + 42
inputset = range(10)
function.compile(inputset, configuration, artifacts, verbose=True, virtual=True)
captured = capsys.readouterr()
assert captured.out.strip() == (
f"""
Computation Graph
------------------------------------------------
{list(artifacts.textual_representations_of_graphs.values())[-1][-1]}
------------------------------------------------
MLIR
------------------------------------------------
Virtual circuits doesn't have MLIR.
------------------------------------------------
Optimizer
------------------------------------------------
Virtual circuits doesn't have optimizer output.
------------------------------------------------
""".strip()
)