Skip to content

Commit 45c8e5a

Browse files
committed
Correct warmup
1 parent fb05135 commit 45c8e5a

File tree

1 file changed

+0
-4
lines changed

1 file changed

+0
-4
lines changed

graph_net/torch/test_compiler.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -194,10 +194,6 @@ def test_single_model(args):
194194
compiled_model_call = lambda: compiled_model(**input_dict)
195195

196196
if args.device == "cuda":
197-
for _ in range(args.warmup):
198-
compiled_model_call()
199-
compiler.synchronize()
200-
201197
eager_times = time_execution_with_cuda_event(
202198
eager_model_call,
203199
num_warmup=args.warmup,

0 commit comments

Comments
 (0)