equal

apache · Oct 10, 2019 · 9752b8e · 9752b8e
1 parent 7fc7c6f
commit 9752b8e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/python/mxnet/numpy_dispatch_protocol.py b/python/mxnet/numpy_dispatch_protocol.py
@@ -162,6 +162,7 @@ def _register_array_function():
     'arctan2',
     'copysign',
     'degrees',
+    'equal',
     'subtract',
     'multiply',
     # Uncomment divide when mxnet.numpy.true_divide is added

diff --git a/tests/python/unittest/test_numpy_interoperability.py b/tests/python/unittest/test_numpy_interoperability.py
@@ -146,6 +146,8 @@ def _prepare_workloads():
     OpArgMngr.add_workload('copysign', np.array([-2, 5, 1, 4, 3], dtype=np.float16), np.array([0, 1, 2, 4, 2], dtype=np.float16))
     OpArgMngr.add_workload('degrees', np.array(np.pi))
     OpArgMngr.add_workload('degrees', np.array(-0.5*np.pi))
+    OpArgMngr.add_workload('equal', np.array([0, 1, 2, 4, 2], dtype=np.float16), np.array([-2, 5, 1, 4, 3], dtype=np.float16))
+    OpArgMngr.add_workload('equal', np.array([np.nan]))
     OpArgMngr.add_workload('subtract', array_pool['4x1'], array_pool['1x2'])
     OpArgMngr.add_workload('subtract', array_pool['4x1'], 2)
     OpArgMngr.add_workload('subtract', 2, array_pool['4x1'])