Skip to content

Commit 666e39d

Browse files
committed
Remove the debugging codes.
1 parent 263709b commit 666e39d

File tree

2 files changed

+3
-8
lines changed

2 files changed

+3
-8
lines changed

graph_net/paddle/test_compiler.py

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -255,14 +255,13 @@ def test_single_model(args):
255255
eager_success = False
256256
try:
257257
print("Run model in eager mode.")
258-
# static_model = get_static_model(args, model)
258+
static_model = get_static_model(args, model)
259259
expected_out, eager_time_stats = measure_performance(
260-
lambda: model(**input_dict), args, synchronizer_func, profile=False
260+
lambda: static_model(**input_dict), args, synchronizer_func, profile=False
261261
)
262262
eager_success = True
263263
except Exception as e:
264264
print(f"Run model in eager mode failed: {str(e)}\n{traceback.format_exc()}")
265-
# sys.exit(0)
266265

267266
# Run on compiling mode
268267
compiled_success = False
@@ -278,9 +277,6 @@ def test_single_model(args):
278277

279278
test_compiler_util.print_running_status(args, eager_success, compiled_success)
280279
if eager_success and compiled_success:
281-
i = 0
282-
# print(f"expected_out[{i}]: {expected_out[i]}")
283-
# print(f"compiled_out[{i}]: {compiled_out[i]}")
284280
check_outputs(args, expected_out, compiled_out)
285281

286282
test_compiler_util.print_times_and_speedup(

graph_net/test_compiler_util.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -186,8 +186,7 @@ def calculate_tolerance_pair(begin, end):
186186

187187

188188
def generate_allclose_configs(cmp_all_close_func):
189-
# tolerance_pair_list = calculate_tolerance_pair(-10, 5)
190-
tolerance_pair_list = calculate_tolerance_pair(-3, 1)
189+
tolerance_pair_list = calculate_tolerance_pair(-10, 5)
191190

192191
cmp_configs = []
193192
for pair in tolerance_pair_list:

0 commit comments

Comments
 (0)