gaspersic/triton
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[TUTORIALS] Removed leftover print

Browse Source
This commit is contained in:
Philippe Tillet
2022-03-28 16:53:23 -07:00
parent e0cc488055
commit bace26143d
1 changed files with 0 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
python/tutorials/01-vector-add.py
View File

@@ -73,7 +73,6 @@ def add(x: torch.Tensor, y: torch.Tensor):
add_kernel[grid](x, y, output, n_elements, time_start, time_end, BLOCK_SIZE=1024) add_kernel[grid](x, y, output, n_elements, time_start, time_end, BLOCK_SIZE=1024)
# We return a handle to z but, since `torch.cuda.synchronize()` hasn't been called, the kernel is still # We return a handle to z but, since `torch.cuda.synchronize()` hasn't been called, the kernel is still
# running asynchronously at this point. # running asynchronously at this point.
print((time_end, time_start))
return output return output