Add an `s5_operator` scan benchmark as an application benchmark
Open
- Dominant language
- C++
- Stars
- 2.5k
- Forks
- 487
- Avg merge
- 2d 7h
- Merged PRs (30d)
- 296
Description
This PR https://github.com/pytorch/pytorch/pull/164859 contains a benchmark for the PyTorch `associative_scan()` using the `s5_operator`:
```Python
def s5_operator(x: torch.Tensor, y: torch.Tensor):
A_i, Bu_i = x
A_j, Bu_j = y
return A_j * A_i, A_j * Bu_i + Bu_j
```
The input to the scan is a tuple of 2d tensors. We should track this as one of the benchmarks we are interested in.
Contributor guide
Assessment
This issue has not been assessed yet.