revert unnecessary changes

triton-lang · Feb 28, 2025 · e58b073 · e58b073
1 parent c404850
commit e58b073
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
@@ -1714,7 +1714,8 @@ def kernel(in_ptr, idx_ptr, out_ptr, shape0, shape1, mask_step, XBLOCK: tl.const
         idx = torch.randint(0, shape1, size=(shape0, shape1), device=device)
 
     val = torch.randn((shape0, shape1), dtype=getattr(torch, dtype_x_str), device=device)
-    dst = torch.zeros((shape0, shape1), dtype=getattr(torch, dtype_x_str), device=device)
+    dst = torch.randn((shape0, shape1), dtype=getattr(torch, dtype_x_str), device=device)
+
     dst_ref = dst.clone()
 
     cnt = 0
@@ -1725,7 +1726,6 @@ def kernel(in_ptr, idx_ptr, out_ptr, shape0, shape1, mask_step, XBLOCK: tl.const
             cnt += 1
 
     kernel[(1, )](val, idx, dst, shape0, shape1, mask_step, 64, num_ctas=num_ctas)
-
     np.testing.assert_allclose(to_numpy(dst_ref), to_numpy(dst), atol=1e-2)