Skip to content

Commit f61e127

Browse files
Shuangping Liufacebook-github-bot
authored andcommitted
Remove fused_params in test_ssd_mixed_kernels_with_vbe (meta-pytorch#3324)
Summary: Pull Request resolved: meta-pytorch#3324 Remove the unnecessary fused param `prefetch_pipeline=true` in the unit test `test_ssd_mixed_kernels_with_vbe`. This is only needed for `UVM_CACHING` to create a multiple-TBE scenario. Here for `KEY_VALUE` SSD TBE this is not needed. Reviewed By: TroyGarden, aliafzal Differential Revision: D81150291 fbshipit-source-id: 33143eeb58a77db43f97f50bb45989b03fb91298
1 parent 55d4ea2 commit f61e127

File tree

1 file changed

+1
-4
lines changed

1 file changed

+1
-4
lines changed

torchrec/distributed/tests/test_model_parallel_nccl_ssd_single_gpu.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -572,9 +572,6 @@ def test_ssd_mixed_kernels_with_vbe(
572572
dtype: DataType,
573573
) -> None:
574574
self._set_table_weights_precision(dtype)
575-
fused_params = {
576-
"prefetch_pipeline": True,
577-
}
578575
constraints = {
579576
table.name: ParameterConstraints(
580577
min_partition=4,
@@ -601,7 +598,7 @@ def test_ssd_mixed_kernels_with_vbe(
601598
embedding_groups={},
602599
tables=self.tables,
603600
# pyre-fixme[6]
604-
sharders=[EmbeddingBagCollectionSharder(fused_params=fused_params)],
601+
sharders=[EmbeddingBagCollectionSharder()],
605602
optim=EmbOptimType.EXACT_SGD,
606603
# The optimizer config here will overwrite the SGD optimizer above
607604
apply_optimizer_in_backward_config={

0 commit comments

Comments
 (0)