Skip to content

Commit 744741c

Browse files
authored
Merge branch 'main' into 3652_retain_grad
2 parents 088e682 + bc79723 commit 744741c

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)