wip

apache · Oct 30, 2024 · b8a39ef · b8a39ef
1 parent f2730a4
commit b8a39ef
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/tests/python/contrib/test_cublas.py b/tests/python/contrib/test_cublas.py
@@ -90,10 +90,14 @@ def verify(target="cuda"):
         a_new[::2] = a_even
         a_new[1::2] = a_odd
         a_new = np.vstack(a_new)
-        a_new = np.vstack([
-            np.vstack([np.vstack(np.hsplit(i, 8)).reshape([4, 32]) for i in np.vsplit(j, N / 4)])
-            for j in np.hsplit(a_new, L / 32)
-        ])
+        a_new = np.vstack(
+            [
+                np.vstack(
+                    [np.vstack(np.hsplit(i, 8)).reshape([4, 32]) for i in np.vsplit(j, N / 4)]
+                )
+                for j in np.hsplit(a_new, L / 32)
+            ]
+        )
         a_new = a_new.reshape([N, L])
         # Transform b to become CUBLASLT_ORDER_COL32 layout
         b_new = np.vstack(