pytorch torchrec issues

pytorch / torchrec

Pytorch domain library for recommendation systems

BSD 3-Clause "New" or "Revised" License

1.78k stars 380 forks source link

issues

Newest

Newest Most commented Recently updated Oldest Least commented Least recently updated

fix missing device in fake operator

#2203 TroyGarden closed 7 hours ago
3
how to mark KJT offsets as dynamic?

#2202 TroyGarden opened 11 hours ago
2
Enable Serving Calibration metric to visualize segment calibration of serving traffic.

#2201 jiayixu64 opened 13 hours ago
1
Enable Serving NE metric to visualize segment NE of serving traffic.

#2200 jiayixu64 opened 13 hours ago
2
fpEBC thrift serialization support

#2199 TroyGarden closed 19 hours ago
3
remove non_strict_export_forward from embedding modules

#2198 TroyGarden closed 11 hours ago
3
serialization to encapsualtion API migration

#2197 TroyGarden closed 15 hours ago
2
revert back to ST state dict for RW sharding

#2196 iamzainhuda closed 1 day ago
1
support dynamic dim for variable lengths in KJT

#2195 TroyGarden closed 12 hours ago
1
add GPU sync tests

#2194 iamzainhuda opened 4 days ago
1
comm_ops split-cat custom_op workaround for unbacked backward

#2193 IvanKobzarev closed 4 days ago
1
force CW shards to be contiguous

#2192 iamzainhuda opened 5 days ago
1
Make promote_high_prefetch_overheaad_table_to_hbm a static function so other proposer can reuse

#2191 levythu closed 4 days ago
1
add callbacks to planner

#2190 iamzainhuda opened 5 days ago
2
Fix Pyre check

#2189 PaulZhang12 opened 5 days ago
1
Fix Unit Test CI Pytest Fixtures

#2188 PaulZhang12 closed 5 days ago
1
KJT.empty_like FX Traceable + Scriptable

#2187 PaulZhang12 closed 5 days ago
1
Fix precommit CI

#2186 PaulZhang12 closed 5 days ago
2
Fix Sigmoid PT2 test to mimic AIMP PT2 and refactor torchbind class register

#2185 PaulZhang12 closed 5 days ago
2
Make LazyAwaitable wait_async dynamo traceable

#2184 IvanKobzarev opened 5 days ago
1
Reland [TorchRec][PT2] KJT custom op for 1d lengths input

#2183 TroyGarden opened 6 days ago
1
Disable non-VB inductor test, until new fbgemm op is not in sigrid release

#2182 IvanKobzarev closed 6 days ago
1
modify the placeholder attr parser to handle dict/list types

#2181 TroyGarden closed 4 days ago
2
comm_ops split-cat custom_op workaround for unbacked backward

#2180 IvanKobzarev closed 6 days ago
2
Add option to pass in mode and func to easy to do strobelight benchmarking

#2179 gnahzg opened 6 days ago
1
Enable non-VB inductor compilation, backward extra numerics checks

#2178 IvanKobzarev closed 6 days ago
1
[Bug][Dynamic Embedding] improper optimizier state_dict `momentum2` key while constructing PSCollection

#2177 JacoCheung opened 1 week ago
1
Split of "[TorchRec][PT2] KJT custom op for 1d lengths input" (#2163)

#2176 TroyGarden closed 1 week ago
2
fix EmbeddingTrainPipelineTest: test_equal_to_non_pipelined

#2175 iamzainhuda closed 6 days ago
1
refactor using encapsulate and decapsulate

#2174 TroyGarden closed 1 day ago
1
fix CI for test_shards_wrapper

#2173 iamzainhuda closed 1 week ago
1
Fix test_pt2 tests

#2172 IvanKobzarev closed 1 week ago
3
Metric State Vectorization

#2171 iamzainhuda opened 1 week ago
1
copy _permute_tensor_by_segments to avoid packaging import error

#2170 joshuadeng opened 1 week ago
1
fix variable assignment in DTensor emb lookup

#2169 iamzainhuda closed 1 week ago
3
Add KeyValueParams

#2168 henrylhtsang closed 1 week ago
4
forward fix embedding_lookup to include DTensor

#2167 iamzainhuda closed 1 week ago
2
An alternative implementation of torchrec serializer

#2166 TroyGarden closed 1 week ago
2
fix device hardcode

#2165 jingsh closed 1 week ago
6
Grouped table generation for unified embedding

#2164 Fiery opened 1 week ago
1
KJT custom op for 1d lengths input

#2163 TroyGarden closed 1 week ago
2
EBC backward numeric correctness tests

#2162 IvanKobzarev closed 1 week ago
1
Remove comm_ops import to workaround torch package issues of lego

#2161 IvanKobzarev closed 1 week ago
1
Refactor source code parity check with `ast`

#2160 XuehaiPan opened 1 week ago
0
benchmark of fbgemm op - regroup_keyed_tensor

#2159 TroyGarden opened 1 week ago
1
benchmark of fbgemm op - permute_multi_embedding

#2158 TroyGarden opened 1 week ago
3
Add GPU trace for KT.regroup benchmark

#2157 TroyGarden closed 1 week ago
2
Back out "QEBC permute order caching"

#2156 gnahzg closed 1 week ago
2
Refactor passing over cache params

#2155 henrylhtsang opened 1 week ago
3
Prevent excess buffer allocation

#2154 Damowerko closed 6 days ago
2