Skip to content

Commit 5ccd9eb

Browse files
authored
Fix elapsed time computation from issue #3649
1 parent 86b1c62 commit 5ccd9eb

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

intermediate_source/torch_compile_tutorial.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -163,7 +163,7 @@ def timed(fn):
163163
result = fn()
164164
end.record()
165165
torch.cuda.synchronize()
166-
return result, start.elapsed_time(end) / 1024
166+
return result, start.elapsed_time(end) / 1000
167167

168168

169169
inp = torch.randn(4096, 4096).cuda()

0 commit comments

Comments
 (0)