fix formatting and fix cuda input_precision

triton-lang · Apr 9, 2024 · fae2f6e · fae2f6e
1 parent 65c9f67
commit fae2f6e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/test/unit/language/test_core.py b/python/test/unit/language/test_core.py
@@ -3183,9 +3183,9 @@ def kernel(X, stride_xm, stride_xk, Y, stride_yk, stride_yn, W, stride_wn, strid
 def test_dot3d(B, num_warps, M, N, K, in_dtype_str, out_dtype_str, device):
     if is_hip():
         # hip does not support tf32 precision, so use ieee for all tests
-        input_precision="ieee"
+        input_precision = "ieee"
     else:
-        input_precision="tf32" if in_dtype_str == 'float32' else "ieee",
+        input_precision = "tf32" if in_dtype_str == 'float32' else "ieee"
 
     @triton.jit
     def kernel(