perf(embedding): use a gather-optimal device layout for nn.Embedding tables - #1
Closed
ani300 wants to merge 1 commit into
Closed
perf(embedding): use a gather-optimal device layout for nn.Embedding tables#1ani300 wants to merge 1 commit into
ani300 wants to merge 1 commit into