yapf

Signed-off-by: Julian Oppermann <julian.oppermann@codeplay.com>
intel · May 20, 2024 · 5dcd09d · 5dcd09d
1 parent 7a29cb3
commit 5dcd09d
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/python/tutorials/09-experimental-block-pointer.py b/python/tutorials/09-experimental-block-pointer.py
@@ -252,9 +252,8 @@ def matmul(a, b, res_dtype):
 
     triton_output = matmul(a, b, res_dtype)
     # torch.matmul clamps values to input dtype; IPEX doesn't support int32 matmul
-    torch_output = torch.matmul(
-        a.to(device='cpu', dtype=res_dtype),
-        b.to(device='cpu', dtype=res_dtype)).to(device='xpu')
+    torch_output = torch.matmul(a.to(device='cpu', dtype=res_dtype), b.to(device='cpu',
+                                                                          dtype=res_dtype)).to(device='xpu')
     print(f"triton_output={triton_output}")
     print(f"torch_output={torch_output}")