Commit f958cab
committed
Update on "[ET-VK] Change weight packing in embedding"
The existing weight tensor for aten.embedding is created using a `tensor_like` from the output tensor, which defaults to channel packed.
However, the weight tensor is actually a 2D-tensor of `(num_embedding, dim_of_embedding)`. It is better in space to use either width or height packing.
This diff changes the implementation to use height-packing.
Differential Revision: [D66421366](https://our.internmc.facebook.com/intern/diff/D66421366/)
[ghstack-poisoned]1 parent c87c6c3 commit f958cab
1 file changed
+1
-1
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
19 | 19 | | |
20 | 20 | | |
21 | 21 | | |
22 | | - | |
23 | 22 | | |
| 23 | + | |
24 | 24 | | |
25 | 25 | | |
26 | 26 | | |
| |||
0 commit comments