lint fix

Author: LeiWang1999

testing/python/jit/test_tilelang_jit_nullptr.py

@@ -7,7 +7,15 @@
 
 
 @tl.jit
-def tensor_null_test(M, N, K, block_M, block_N, block_K, dtype="float16", accum_dtype="float", with_bias=False):
+def tensor_null_test(M,
+                     N,
+                     K,
+                     block_M,
+                     block_N,
+                     block_K,
+                     dtype="float16",
+                     accum_dtype="float",
+                     with_bias=False):
 
     @T.prim_func
     def main(
@@ -44,9 +52,11 @@ def run_test(M, N, K, block_M, block_N, block_K, dtype="float16", accum_dtype="f
     a = torch.randn(M, K, device="cuda", dtype=map_torch_type(dtype))
     b = torch.randn(N, K, device="cuda", dtype=map_torch_type(dtype))
     c = torch.zeros(M, N, device="cuda", dtype=map_torch_type(accum_dtype))
-    kernel = tensor_null_test(M, N, K, block_M, block_N, block_K, dtype, accum_dtype, with_bias=False)
+    kernel = tensor_null_test(
+        M, N, K, block_M, block_N, block_K, dtype, accum_dtype, with_bias=False)
     kernel(a, b, c, None)
 
+
 def test_nullptr():
     run_test(1024, 1024, 1024, 128, 128, 32)