Skip to content

Commit

Permalink
Support 2D inner reduction scheduler with autotuning (#3456)
Browse files Browse the repository at this point in the history
# 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
rdspring1 authored Dec 11, 2024
1 parent b53c103 commit 1b29928
Show file tree
Hide file tree
Showing 2 changed files with 418 additions and 0 deletions.
Loading

0 comments on commit 1b29928

Please sign in to comment.