We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 11a435a commit fc865beCopy full SHA for fc865be
graph_net/paddle/test_compiler.py
@@ -112,6 +112,7 @@ def test_single_model(args):
112
print("-- Run with compiled mode")
113
build_strategy = paddle.static.BuildStrategy()
114
# build_strategy.build_cinn_pass = True
115
+ compilation_start_time = time.time()
116
compiled_model = paddle.jit.to_static(
117
model_dy,
118
input_spec=input_spec,
@@ -148,7 +149,7 @@ def test_single_model(args):
148
149
raise ValueError("Illegal return value.")
150
151
def print_cmp(key, func, **kwargs):
- cmp_ret = func(processed_expected_out, processed_compiled_out, **kwargs)
152
+ cmp_ret = func(expected_out, compiled_out, **kwargs)
153
print(
154
f"{args.log_prompt} {key} model_path:{args.model_path} {cmp_ret}",
155
file=sys.stderr,
0 commit comments