torchrec

torchrec copied to clipboard

Add host_port_pair in ShardMetadata to facilitate uneven rowwise sharding testing

3

Summary: DI's test on uneven sharding use host port pair. This diff add this info in the ShardMetadata to enable this. https://fburl.com/gdoc/r8qeeb96 Differential Revision: D50977976

gnahzg

Allow Missing Labels for MTML models

3

Differential Revision: D50273422

PaulZhang12

take in ezyang changes and test dlrm script

3

Differential Revision: D50575605

YLGH

merge atm and no atm quantization

1

Summary: merge embedding quantization method of different path Reviewed By: strisunshinewentingwang Differential Revision: D50432952

SeanXiaohengMao

Dlrm blaze checkpoint5 import

1

YLGH

prop async tensor

1

Differential Revision: D50471106

YLGH

Add uvm caching with prefetch test for group_tables

1

Summary: Adding a test after enable embedding dim bucketizer diff. Differential Revision: D50483066

henrylhtsang

Add tests for group_tables in hbm cases

2

Summary: Adding a unit test for group_tables. This way of testing isn't very reliable, since it just look at how the tables are grouped. Also it relies on ordering of...

henrylhtsang

Inference docker does not support torchrec 0.5.0?

1

Hi all, I'm facing an issue with the sahanp465/multipy Docker image, as it only supports CUDA 11.3, but torchrec 0.5.0 requires at least CUDA 11.8. I'm wondering what the recommended...

tiankongdeguiji

upgrade pyre version in `fbcode/torchrec` - batch 1

1

Differential Revision: D49425673

henrylhtsang