[tst] explicitly specify gather mode in roofline test

jakevdp · jakevdp · commit 4a5049a42b1c · 2025-11-21T14:26:51.000-08:00
diff --git a/tests/roofline_test.py b/tests/roofline_test.py
@@ -965,6 +965,7 @@ def test_gather_batching_dims_roofline(self):
         y,
         dimension_numbers=dimension_numbers,
         slice_sizes=(1, 1, 3),
+        mode='promise_in_bounds',
     )
 
     _, result = roofline.roofline(f)(operand, indices)

Original file line number	Diff line number	Diff line change
`@@ -965,6 +965,7 @@ def test_gather_batching_dims_roofline(self):`
`965`	`965`	`y,`
`966`	`966`	`dimension_numbers=dimension_numbers,`
`967`	`967`	`slice_sizes=(1, 1, 3),`
	`968`	`+ mode='promise_in_bounds',`
`968`	`969`	`)`
`969`	`970`
`970`	`971`	`_, result = roofline.roofline(f)(operand, indices)`