How to obtain arithmetic intensity data for the Pallas kernel? (For rooflines)

[Prayer3th]

Hey guys, we are currently using rooflines to identify where the bottlenecks in inference performance lie.
However, when using xprof or similar tools to obtain actual arithmetic intensity data for the Pallas kernel, the results consistently show as 0.00 or Unknown. Are there any better methods for acquiring this data?
(The RPA-related data in the figure refers to the ragged_paged_attention kernel.)