Commit 33c867f
authored
perf: reduce pre-GEMM overhead in einsum2 prepare path (#95)
* Optimize ND kernel fallback by replacing recursion with iterative carry loop
* fmt: apply rustfmt for iterative ND kernel test
* perf: reduce pre-gemm overhead in einsum2 prepare path1 parent 8e6cab2 commit 33c867f
5 files changed
Lines changed: 243 additions & 45 deletions
File tree
- strided-einsum2/src
- strided-opteinsum/src
- strided-traits/src
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
26 | 26 | | |
27 | 27 | | |
28 | 28 | | |
29 | | - | |
| 29 | + | |
30 | 30 | | |
31 | 31 | | |
32 | 32 | | |
| |||
0 commit comments