Add asymmetric shapes to test sum Triton kernels

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 asymmetric shapes to test sum Triton kernels #2310

Closed jananisriram closed 2 weeks ago

jananisriram commented 2 weeks ago

Summary: Add asymmetric shapes to get_input_iter() in order to test accuracy and performance of sum Triton kernel implementations against PyTorch.

This diff generates tensors with dimensions of different sizes. For example, a 2D asymmetric tensor would have shape (n, n + 3); a 3D asymmetric tensor would have shape (n, n + 3, n + 6).

Reviewed By: jbschlosser

Differential Revision: D58509022

facebook-github-bot commented 2 weeks ago

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

facebook-github-bot commented 2 weeks ago

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