more newlines

triton-lang · Nov 26, 2024 · 8f1358e · 8f1358e
1 parent 6a2f788
commit 8f1358e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/python/test/unit/language/test_core.py b/python/test/unit/language/test_core.py
@@ -6129,4 +6129,4 @@ def triton_gather(src: torch.Tensor, axis: int, indices: torch.Tensor):
     indices = torch.randint(0, src.shape[axis], indices_shape, device='cuda')
     ref = torch.gather(src, axis, indices)
     result = triton_gather(src, axis, indices)
-    assert torch.all(ref == result)
+    assert torch.all(ref == result)
diff --git a/test/Conversion/triton_to_tritongpu.mlir b/test/Conversion/triton_to_tritongpu.mlir
@@ -118,4 +118,4 @@ tt.func @gather_op() {
   // CHECK: tt.gather %{{.*}}[%{{.*}}] {axis = 0 : i32} : (tensor<128x4xf32, #blocked>, tensor<256x4xi32, #blocked>) -> tensor<256x4xf32, #blocked>
   %0 = tt.gather %cst[%cst_0] {axis = 0 : i32} : (tensor<128x4xf32>, tensor<256x4xi32>) -> tensor<256x4xf32>
   tt.return
-}
+}