Skip to content

Kittens-like on Aster 22/n - Add scf.for rotation to move compute fir…#424

Draft
nicolasvasilache wants to merge 1 commit intomainfrom
kit-perf-015
Draft

Kittens-like on Aster 22/n - Add scf.for rotation to move compute fir…#424
nicolasvasilache wants to merge 1 commit intomainfrom
kit-perf-015

Conversation

@nicolasvasilache
Copy link
Collaborator

…st in the loop so we can hide stuff behind

Repro:

python contrib/kittens/test/test_perf_001_gemm_fp16_weak_scaled.py  --m-wg 1 --n-wg 1 --m-waves 1 --n-waves 1  --m-tiles-wg 2 --n-tiles-wg 2 --k-tiles 1  --stages 3  -
-print-ir-after-all

…st in the loop so we can hide stuff behind

Repro:

```
python contrib/kittens/test/test_perf_001_gemm_fp16_weak_scaled.py  --m-wg 1 --n-wg 1 --m-waves 1 --n-waves 1  --m-tiles-wg 2 --n-tiles-wg 2 --k-tiles 1  --stages 3  -
-print-ir-after-all
```

Signed-off-by: Nicolas Vasilache <nicolasvasilache@users.noreply.github.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant