Skip to content

Commit 01b107e

Browse files
committed
Fix
1 parent 9370714 commit 01b107e

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

test/legacy_test/test_activation_op.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -168,6 +168,11 @@ def init_shape(self):
168168
self.shape = []
169169

170170

171+
class TestExpPrim_ZeroSize(TestExpFp32_Prim):
172+
def init_shape(self):
173+
self.shape = [0, 2]
174+
175+
171176
class TestExp_Complex64(OpTest):
172177
def setUp(self):
173178
self.op_type = "exp"
@@ -314,6 +319,11 @@ def init_shape(self):
314319
self.shape = []
315320

316321

322+
class TestExpm1_ZeroSize(TestExpm1):
323+
def init_shape(self):
324+
self.shape = [0, 2]
325+
326+
317327
class TestExpm1API(unittest.TestCase):
318328
def init_dtype(self):
319329
self.dtype = 'float64'

0 commit comments

Comments
 (0)