diff --git a/python/tutorials/01-vector-add.py b/python/tutorials/01-vector-add.py
index 1e77ca7c1d..e527e5fc7a 100644
--- a/python/tutorials/01-vector-add.py
+++ b/python/tutorials/01-vector-add.py
@@ -87,10 +87,10 @@ def add(x: torch.Tensor, y: torch.Tensor):
 y = torch.rand(size, device=DEVICE)
 output_torch = x + y
 output_triton = add(x, y)
-print(output_torch.cpu())
-print(output_triton.cpu())
+print(output_torch)
+print(output_triton)
 print(f'The maximum difference between torch and triton is '
-      f'{torch.max(torch.abs(output_torch.cpu() - output_triton.cpu()))}')
+      f'{torch.max(torch.abs(output_torch - output_triton))}')
 
 # %%
 # Seems like we're good to go!