test_comm test inductor (#2017)

Summary: Start testing inductor backend for collective ops instead of just eager Differential Revision: D57558437
pytorch · May 20, 2024 · 3718689 · 3718689
1 parent 889878c
commit 3718689
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/torchrec/distributed/tests/test_comm.py b/torchrec/distributed/tests/test_comm.py
@@ -138,7 +138,7 @@ def fn(*args, **kwargs) -> List[torch.Tensor]:
     )
     # pyre-ignore
     @given(
-        torch_compile_args=st.sampled_from([None, ("eager", True)]),
+        torch_compile_args=st.sampled_from([None, ("inductor", True)]),
         specify_pg=st.sampled_from([True]),
     )
     @settings(deadline=None)
@@ -263,7 +263,7 @@ def fn(*args, **kwargs) -> torch.Tensor:
     )
     # pyre-ignore
     @given(
-        torch_compile_args=st.sampled_from([None, ("eager", True)]),
+        torch_compile_args=st.sampled_from([None, ("inductor", True)]),
         specify_pg=st.sampled_from([True]),
     )
     @settings(deadline=None)
@@ -378,7 +378,7 @@ def fn(*args, **kwargs) -> torch.Tensor:
     )
     # pyre-ignore
     @given(
-        torch_compile_args=st.sampled_from([None, ("eager", True)]),
+        torch_compile_args=st.sampled_from([None, ("inductor", True)]),
         specify_pg=st.sampled_from([True]),
     )
     @settings(deadline=None)
@@ -454,7 +454,7 @@ def fn(*args, **kwargs) -> torch.Tensor:
     )
     # pyre-ignore
     @given(
-        torch_compile_args=st.sampled_from([None, ("eager", True)]),
+        torch_compile_args=st.sampled_from([None, ("inductor", True)]),
         specify_pg=st.sampled_from([True]),
     )
     @settings(deadline=None)
@@ -541,7 +541,7 @@ def fn(*args, **kwargs) -> torch.Tensor:
     )
     # pyre-ignore
     @given(
-        torch_compile_args=st.sampled_from([None, ("eager", True)]),
+        torch_compile_args=st.sampled_from([None, ("inductor", True)]),
         specify_pg=st.sampled_from([True]),
     )
     @settings(deadline=None)
@@ -618,7 +618,7 @@ def fn(*args, **kwargs) -> torch.Tensor:
     )
     # pyre-ignore
     @given(
-        torch_compile_args=st.sampled_from([None, ("eager", True)]),
+        torch_compile_args=st.sampled_from([None, ("inductor", True)]),
         specify_pg=st.sampled_from([True]),
     )
     @settings(deadline=None)
@@ -683,7 +683,7 @@ def fn(*args, **kwargs) -> torch.Tensor:
     )
     # pyre-ignore
     @given(
-        torch_compile_args=st.sampled_from([None, ("eager", True)]),
+        torch_compile_args=st.sampled_from([None, ("inductor", True)]),
         specify_pg=st.sampled_from([True]),
     )
     @settings(deadline=None)