torchrec

torchrec copied to clipboard

Fix ParameterConstraints docstrings

2

Summary: Fix horrible docstrings (created by me) on pytorch.org right now. Reason is for dataclasses, we need to handle the docs carefully. Before: {F1644093577} same attribute would appear twice. After:...

henrylhtsang

Formatting

1

Summary: arc lint external OSS PR Differential Revision: D57507551

PaulZhang12

test_comm test inductor

2

Summary: Start testing inductor backend for collective ops instead of just eager Differential Revision: D57558437

IvanKobzarev

Fix OSS pyre

2

Summary: OSS CI Pyre lint failing Differential Revision: D57350370

sarckk

Remove extremely slow logic from RequestOnlySplitEmbeddingBagCollectionsSparseArch

2

Differential Revision: D57290467

PaulZhang12

Convert torchrec to fast by default architecture] [batch:74/79]

3

Differential Revision: D57262910

maggiemoss

Convert .pyre_configuration.local to fast by default architecture] [batch:51/51]

2

Reviewed By: connernilsen Differential Revision: D57271924

connernilsen

block_bucketize_sparse_features cannot be parallelized with the forward pass

zhaozheng09

add l2_cache_size config path from model config

3

Summary: support l2 cache size as a config passed from model side Differential Revision: D61418000

duduyi2013

TorchRec MX4 padding support

8

Summary: rowwise padding is needed to fix NE of Ads ranking models for AlltoAll communication with MX4. Differential Revision: D61928939

qchip