Function np_bf162np_float

tests/python/codegen/test_target_codegen_cuda.py:125–129

Convert a numpy array of bf16 (uint16) to a numpy array of float

(arr)

        return np.right_shift(orig + bias, 16).astype("uint16")

    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)
        return u32.view("<f4")

    def check_cuda(n, lanes):
        vec_dtype = f"bfloat16x{lanes}"