Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Support 2D inner reduction scheduler with autotuning (#3456)
# Summary: This PR creates an SOL autotuning script for the 2d inner reduction scheduler. It trains a random forest to predict the best performing configuration for the reduction scheduler. # Inner Reduction Fusions: 1. Sum --- `y = sum(x, dim=-1)` 2. Add Sum --- `z = sum(x1 + x2 + x3 + x4, dim=-1)` 3. Tanh Sum --- `y = sum(tanh(x), dim=-1)` 4. Exp Sum --- `z = sum(exp(x), dim=-1)`
- Loading branch information