benchmark

31ebbaaa - Refactor jagged operator benchmarks in TritonBench with helper functions for readability

Refactor jagged operator benchmarks in TritonBench with helper functions for readability Summary: Refactor `jagged_sum`, `jagged_mean`, `jagged_softmax`, and `jagged_layer_norm` TritonBench benchmarks with existing and new helper functions in `jagged_utils.py`. This diff improves readability of existing TritonBench code and adds helper functions, to be used across multiple jagged operators, to `jagged_utils.py`. Reviewed By: xuzhao9, davidberard98 Differential Revision: D60065821 fbshipit-source-id: 6b870743cc0b053b54057cf9dc3c72966566d51c