Skip to content

Commit 761093e

Browse files
authored
Remove clone data in PyTorch profiling (#117)
1 parent 2beb3ed commit 761093e

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

problems/amd_202602/eval.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -319,7 +319,7 @@ def run_single_profile(test: TestCase) -> str:
319319
torch.cuda.synchronize()
320320

321321
with profile(activities=[ProfilerActivity.CPU, ProfilerActivity.CUDA]) as prof:
322-
submission_output = custom_kernel(_clone_data(data))
322+
submission_output = custom_kernel(data)
323323
torch.cuda.synchronize()
324324
return prof.key_averages().table(sort_by="self_cuda_time_total", row_limit=20)
325325

0 commit comments

Comments
 (0)