Update test_mul.py

complex test
Oneflow-Inc · Jun 29, 2023 · f551bfe · f551bfe
1 parent 93e2000
commit f551bfe
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/oneflow/test/modules/test_mul.py b/python/oneflow/test/modules/test_mul.py
@@ -209,7 +209,7 @@ def test_broadcast_mul(test_case):
         x.mul_(y)
         return x
 
-    @autotest(n=6)
+    @autotest(n=6, include_complex=True)
     def test_non_contiguous_inplace_mul(test_case):
         device = random_device()
         x = random_tensor(2, 2, 4).to(device)
@@ -218,7 +218,7 @@ def test_non_contiguous_inplace_mul(test_case):
         y *= random_tensor(2, 2, 2).to(device)
         return y
 
-    @autotest(n=10)
+    @autotest(n=10, include_complex=True)
     def test_scalar_mul_with_random_devices(test_case):
         x1_device = random_device()
         x2_device = random_device()