Replacing `is` with `==` for the dtype check.
authorPrashant Kumar <prashant@nod-labs.com>
Thu, 1 Dec 2022 16:27:33 +0000 (16:27 +0000)
committerPrashant Kumar <prashant@nod-labs.com>
Fri, 2 Dec 2022 02:34:31 +0000 (02:34 +0000)
commit58c8b253cdd5ccd4e61d5854a8a614b35498276f
treedeb1a894e7095953e258fb5f9b277bcbebe8cb9f
parent0fb74d0ff828e817caf92e83c6b1f25a8fd491ad
Replacing `is` with `==` for the dtype check.

>>> a = np.ndarray([1,1]).astype(np.half)
>>> a
array([[0.007812]], dtype=float16)
>>> a.dtype
dtype('float16')
>>> a.dtype == np.half
True
>>> a.dtype == np.float16
True
>>> a.dtype is np.float16
False

Checking with `is` leads to inconsistency in checking.

Reviewed By: silvas

Differential Revision: https://reviews.llvm.org/D139121
mlir/python/mlir/runtime/np_to_memref.py