search

pytorch / benchmark

TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.
BSD 3-Clause "New" or "Revised" License
817 stars 259 forks source link

Add plots variable along 1 dimension to sum operator in TritonBench #2330

Closed jananisriram closed 6 days ago

jananisriram commented 1 week ago

Summary: Add basic plotting functionality to the sum operator benchmark in TritonBench, graphing shape in 1 dimension against the latency of the PyTorch and Triton kernel sum benchmarks. This diff assumes that if a plot is requested, all dimensions except for one have their sizes specified in command line arguments, where the x-axis plots the different sizes of the singular variable dimension. This diff references the gemm operator plotting function and makes it easier to interpret the performance of different Triton kernels against a baseline PyTorch implementation.

Reviewed By: YuqingJ

Differential Revision: D58680878

facebook-github-bot commented 1 week ago

This pull request was exported from Phabricator. Differential Revision: D58680878

facebook-github-bot commented 1 week ago

This pull request was exported from Phabricator. Differential Revision: D58680878

facebook-github-bot commented 1 week ago

This pull request was exported from Phabricator. Differential Revision: D58680878

facebook-github-bot commented 6 days ago

This pull request was exported from Phabricator. Differential Revision: D58680878

facebook-github-bot commented 6 days ago

This pull request has been merged in pytorch/benchmark@2643d15a74757803d36285c61d2c058b7d808a24.