Commit 145ba58
1 parent 26e0732 commit 145ba58
File tree
6 files changed
+100
-16
lines changed- torchrec
- distributed
- test_utils
- tests
- train_pipeline/tests
- modules
6 files changed
+100
-16
lines changedDiff for: torchrec/distributed/embedding.py
+11-2
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
26 | 26 |
| |
27 | 27 |
| |
28 | 28 |
| |
| 29 | + | |
29 | 30 |
| |
30 | 31 |
| |
31 | 32 |
| |
| |||
90 | 91 |
| |
91 | 92 |
| |
92 | 93 |
| |
| 94 | + | |
93 | 95 |
| |
94 | 96 |
| |
95 | 97 |
| |
| |||
1198 | 1200 |
| |
1199 | 1201 |
| |
1200 | 1202 |
| |
1201 |
| - | |
| 1203 | + | |
1202 | 1204 |
| |
| 1205 | + | |
| 1206 | + | |
| 1207 | + | |
| 1208 | + | |
| 1209 | + | |
| 1210 | + | |
| 1211 | + | |
1203 | 1212 |
| |
1204 | 1213 |
| |
1205 | 1214 |
| |
| |||
1209 | 1218 |
| |
1210 | 1219 |
| |
1211 | 1220 |
| |
1212 |
| - | |
| 1221 | + | |
1213 | 1222 |
| |
1214 | 1223 |
| |
1215 | 1224 |
| |
|
Diff for: torchrec/distributed/embeddingbag.py
+12-4
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
27 | 27 |
| |
28 | 28 |
| |
29 | 29 |
| |
| 30 | + | |
30 | 31 |
| |
31 | 32 |
| |
32 | 33 |
| |
| |||
94 | 95 |
| |
95 | 96 |
| |
96 | 97 |
| |
| 98 | + | |
97 | 99 |
| |
98 | 100 |
| |
99 | 101 |
| |
| |||
656 | 658 |
| |
657 | 659 |
| |
658 | 660 |
| |
659 |
| - | |
660 |
| - | |
661 |
| - | |
| 661 | + | |
662 | 662 |
| |
663 | 663 |
| |
664 | 664 |
| |
| |||
1189 | 1189 |
| |
1190 | 1190 |
| |
1191 | 1191 |
| |
1192 |
| - | |
| 1192 | + | |
| 1193 | + | |
| 1194 | + | |
1193 | 1195 |
| |
| 1196 | + | |
| 1197 | + | |
| 1198 | + | |
| 1199 | + | |
| 1200 | + | |
| 1201 | + | |
1194 | 1202 |
| |
1195 | 1203 |
| |
1196 | 1204 |
| |
|
+26-6
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
147 | 147 |
| |
148 | 148 |
| |
149 | 149 |
| |
| 150 | + | |
150 | 151 |
| |
151 | 152 |
| |
152 | 153 |
| |
| |||
177 | 178 |
| |
178 | 179 |
| |
179 | 180 |
| |
180 |
| - | |
181 |
| - | |
182 |
| - | |
| 181 | + | |
| 182 | + | |
| 183 | + | |
183 | 184 |
| |
184 | 185 |
| |
185 | 186 |
| |
| |||
188 | 189 |
| |
189 | 190 |
| |
190 | 191 |
| |
191 |
| - | |
192 |
| - | |
| 192 | + | |
| 193 | + | |
| 194 | + | |
| 195 | + | |
| 196 | + | |
| 197 | + | |
| 198 | + | |
| 199 | + | |
| 200 | + | |
| 201 | + | |
| 202 | + | |
| 203 | + | |
| 204 | + | |
| 205 | + | |
| 206 | + | |
| 207 | + | |
| 208 | + | |
| 209 | + | |
| 210 | + | |
| 211 | + | |
193 | 212 |
| |
194 | 213 |
| |
195 | 214 |
| |
| |||
200 | 219 |
| |
201 | 220 |
| |
202 | 221 |
| |
203 |
| - | |
204 | 222 |
| |
205 | 223 |
| |
206 | 224 |
| |
| |||
297 | 315 |
| |
298 | 316 |
| |
299 | 317 |
| |
| 318 | + | |
300 | 319 |
| |
301 | 320 |
| |
302 | 321 |
| |
| |||
319 | 338 |
| |
320 | 339 |
| |
321 | 340 |
| |
| 341 | + | |
322 | 342 |
| |
323 | 343 |
| |
324 | 344 |
| |
|
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
376 | 376 |
| |
377 | 377 |
| |
378 | 378 |
| |
| 379 | + | |
| 380 | + | |
| 381 | + | |
| 382 | + | |
| 383 | + | |
| 384 | + | |
| 385 | + | |
| 386 | + | |
| 387 | + | |
| 388 | + | |
| 389 | + | |
| 390 | + | |
| 391 | + | |
| 392 | + | |
| 393 | + | |
| 394 | + | |
| 395 | + | |
| 396 | + | |
| 397 | + | |
| 398 | + | |
| 399 | + | |
| 400 | + | |
| 401 | + | |
| 402 | + | |
| 403 | + | |
| 404 | + | |
| 405 | + | |
| 406 | + | |
| 407 | + | |
| 408 | + | |
| 409 | + | |
| 410 | + | |
| 411 | + | |
| 412 | + | |
| 413 | + | |
| 414 | + | |
| 415 | + | |
| 416 | + | |
| 417 | + | |
| 418 | + | |
| 419 | + |
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
160 | 160 |
| |
161 | 161 |
| |
162 | 162 |
| |
163 |
| - | |
| 163 | + | |
164 | 164 |
| |
165 | 165 |
| |
166 | 166 |
| |
| |||
169 | 169 |
| |
170 | 170 |
| |
171 | 171 |
| |
172 |
| - | |
| 172 | + | |
173 | 173 |
| |
174 | 174 |
| |
175 | 175 |
| |
|
Diff for: torchrec/modules/embedding_modules.py
+8-2
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
19 | 19 |
| |
20 | 20 |
| |
21 | 21 |
| |
| 22 | + | |
22 | 23 |
| |
23 | 24 |
| |
24 | 25 |
| |
| |||
218 | 219 |
| |
219 | 220 |
| |
220 | 221 |
| |
221 |
| - | |
| 222 | + | |
| 223 | + | |
| 224 | + | |
| 225 | + | |
222 | 226 |
| |
223 | 227 |
| |
224 | 228 |
| |
| |||
229 | 233 |
| |
230 | 234 |
| |
231 | 235 |
| |
| 236 | + | |
232 | 237 |
| |
233 | 238 |
| |
234 | 239 |
| |
| |||
448 | 453 |
| |
449 | 454 |
| |
450 | 455 |
| |
451 |
| - | |
| 456 | + | |
452 | 457 |
| |
453 | 458 |
| |
454 | 459 |
| |
| |||
461 | 466 |
| |
462 | 467 |
| |
463 | 468 |
| |
| 469 | + | |
464 | 470 |
| |
465 | 471 |
| |
466 | 472 |
| |
|
0 commit comments