[test] explicitly specify gather mode in roofline test#33483

jakevdp wants to merge 1 commit intojax-ml:mainjax-ml/jax:mainfrom

jakevdp:roofline-testjakevdp/jax:roofline-testCopy head branch name to clipboard