liangfu commented on a change in pull request #5601:
URL: https://github.com/apache/incubator-tvm/pull/5601#discussion_r427691578
##########
File path: tests/python/unittest/test_target_codegen_llvm.py
##########
@@ -710,6 +710,52 @@ def _transform(f, *_):
module(a_, b_, c_)
tvm.testing.assert_allclose(c_.asnumpy(), (a_.asnumpy() *
2).astype('int32'))
+def np_float2np_bf16(arr):
+ ''' Convert a numpy array of float to a numpy array
+ of bf16 in uint16'''
+ orig = arr.view('<u4')
+ bias = np.bitwise_and(np.right_shift(orig, 16), 1) + 0x7FFF
+ return np.right_shift(orig + bias, 16).astype('uint16')
+
+def np_float2tvm_bf16(arr):
+ ''' Convert a numpy array of float to a TVM array
+ of bf16'''
+ nparr = np_float2np_bf16(arr)
+ return tvm.nd.empty(nparr.shape, 'bf16').copyfrom(nparr)
+
+def np_bf162np_float(arr):
+ ''' Convert a numpy array of bf16 (uint16) to a numpy array
+ of float'''
+ u32 = np.left_shift(arr.astype('uint32'), 16)
Review comment:
Can we reused the following code snippet, which preserves endianness
checks?
https://github.com/apache/incubator-tvm/blob/6cbda80227fc18a859c4b01f57f75abbd7a16181/3rdparty/bfloat16/bfloat16.cc#L27-L35
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
us...@infra.apache.org
