add test case (DeepLink-org#286)

* clamp,clamp_max,clamp_min * where * tanh * tril * permute * pad * prod * cumsum * cdist * bitwise_not * reciprocal * bernoulli * mm * roll * unfold * expand * copy_ * lint * device_config * device config * bernoulli
DoorKickers · Oct 8, 2023 · d9a801b · d9a801b
1 parent 4c20dd5
commit d9a801b
Show file tree

Hide file tree

Showing 5 changed files with 759 additions and 120 deletions.
diff --git a/diopi_test/python/conformance/conformance_test.py b/diopi_test/python/conformance/conformance_test.py
@@ -214,7 +214,8 @@ def test_uniform(input, start=0, end=1):
 
     def test_bernoulli(input, inplace=False, p=None):
         p_numpy = input.numpy()
-        p = p_numpy.mean() if p is None else p
+        if input.numel() > 0:
+            p = p_numpy.mean() if p is None else p
         state = build_generator_state(input.context())
         generator = Generator(state)
         out = F.bernoulli(input, inplace, p, generator)