Only run WeightedMultiSample test on CPU and not GPU.

go2newera · Mar 20, 2018 · 4d5ecf9 · 4d5ecf9
1 parent 12fc331
commit 4d5ecf9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/caffe2/python/operator_test/weighted_multi_sample_test.py b/caffe2/python/operator_test/weighted_multi_sample_test.py
@@ -17,7 +17,7 @@ class TestWeightedMultiSample(hu.HypothesisTestCase):
     @given(
         num_samples=st.integers(min_value=0, max_value=128),
         data_len=st.integers(min_value=0, max_value=10000),
-        **hu.gcs
+        **hu.gcs_cpu_only
     )
     def test_weighted_multi_sample(self, num_samples, data_len, gc, dc):
         weights = np.zeros((data_len))