Skip to content

Commit 528aa20

Browse files
committed
cleanup formatting in division test
1 parent 74c8e78 commit 528aa20

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

python/test/regression/test_divide.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -40,9 +40,9 @@ def divide_kernel(a, b, out_ptr0, out_ptr1, out_ptr2, out_ptr3, out_ptr4, xnumel
4040

4141
torch.manual_seed(0)
4242

43-
outputs_float_div = "tl.store(out_ptr0 + (x0), tmp4, xmask)\n tl.store(out_ptr3 + (x0), tmp4, xmask)" if float_div is True else ""
44-
outputs_floor = "\n tl.store(out_ptr1 + (x0), tmp5, xmask)\n tl.store(out_ptr4 + (x0), tmp5, xmask)" if floor is True else ""
45-
outputs_trunc = "\n tl.store(out_ptr2 + (x0), tmp6, xmask)" if trunc is True else ""
43+
outputs_float_div = "tl.store(out_ptr0 + (x0), tmp4, xmask)\n tl.store(out_ptr3 + (x0), tmp4, xmask)" if float_div else ""
44+
outputs_floor = " tl.store(out_ptr1 + (x0), tmp5, xmask)\n tl.store(out_ptr4 + (x0), tmp5, xmask)" if floor else ""
45+
outputs_trunc = " tl.store(out_ptr2 + (x0), tmp6, xmask)" if trunc else ""
4646

4747
divide_kernel = patch_kernel(divide_kernel,
4848
{"GENERATE_OUTPUTS_HERE": f"{outputs_float_div}\n{outputs_floor}\n{outputs_trunc}"})

0 commit comments

Comments
 (0)