Menooker commented on a change in pull request #5601:
URL: https://github.com/apache/incubator-tvm/pull/5601#discussion_r427698163
##########
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:
If my understanding is correct, we don't need to care about endianness.
BF16 conversions only involves getting higher-ordered bits. And the operation
to get higher-ordered bits in C++/Numpy is well-defined.
----------------------------------------------------------------
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
