Commit c490b9f
Adds baseline implementation for MetaShuffling and some cleanups. (#4080)
Summary:
X-link: facebookresearch/FBGEMM#1164
Adds baseline implementation for MetaShuffling and some cleanups.
Differential Revision: D741010691 parent e7d98ca commit c490b9f
File tree
4 files changed
+389
-127
lines changed- fbgemm_gpu/experimental/gen_ai
- gen_ai/moe
- test/moe
4 files changed
+389
-127
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
59 | 59 | | |
60 | 60 | | |
61 | 61 | | |
62 | | - | |
| 62 | + | |
63 | 63 | | |
64 | 64 | | |
65 | 65 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
1 | 1 | | |
2 | 2 | | |
3 | | - | |
| 3 | + | |
4 | 4 | | |
5 | | - | |
| 5 | + | |
6 | 6 | | |
7 | | - | |
| 7 | + | |
8 | 8 | | |
9 | 9 | | |
| 10 | + | |
| 11 | + | |
| 12 | + | |
| 13 | + | |
| 14 | + | |
| 15 | + | |
0 commit comments