Add fused compute kernel to PT2 multiprocess test (#2235)

Summary: Pull Request resolved: #2235 As titled Reviewed By: TroyGarden Differential Revision: D59872821 fbshipit-source-id: 241d09e3d5836803fca6734553c76b567b57958f
pytorch · Jul 19, 2024 · a68a99f · a68a99f
1 parent 50ecc5c
commit a68a99f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/torchrec/distributed/tests/test_pt2_multiprocess.py b/torchrec/distributed/tests/test_pt2_multiprocess.py
@@ -509,6 +509,7 @@ def disable_cuda_tf32(self) -> bool:
         kernel_type=st.sampled_from(
             [
                 EmbeddingComputeKernel.DENSE.value,
+                EmbeddingComputeKernel.FUSED.value,
             ],
         ),
         given_config_tuple=st.sampled_from(