Merge remote-tracking branch 'origin/main' into etiotto.raise_block_p…

…tr.8
intel · Jan 24, 2025 · c77d178 · c77d178
2 parents f3233bd + b018ed6
commit c77d178
Showing 1 changed file with 3 additions and 3 deletions.
diff --git 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!