[BugFix]Ensure that bf16 arrays are created as expected (#16436)

sisleyli · Bin Li · web-flow · commit ccca00a07811 · 2024-01-21T05:07:40.000+09:00
Co-authored-by: Bin Li &lt;binli1@amd.com&gt;
diff --git a/python/tvm/runtime/ndarray.py b/python/tvm/runtime/ndarray.py
@@ -176,6 +176,8 @@ def copyfrom(self, source_array):
         if (not source_array.flags["C_CONTIGUOUS"]) or (
             dtype == "bfloat16" or dtype != np_dtype_str
         ):
+            if dtype == "bfloat16":
+                source_array = np.frombuffer(source_array.tobytes(), "uint16")
             source_array = np.ascontiguousarray(
                 source_array, dtype="uint16" if dtype == "bfloat16" else dtype
             )

Original file line number	Diff line number	Diff line change
`@@ -176,6 +176,8 @@ def copyfrom(self, source_array):`
`176`	`176`	`if (not source_array.flags["C_CONTIGUOUS"]) or (`
`177`	`177`	`dtype == "bfloat16" or dtype != np_dtype_str`
`178`	`178`	`):`
	`179`	`+ if dtype == "bfloat16":`
	`180`	`+ source_array = np.frombuffer(source_array.tobytes(), "uint16")`
`179`	`181`	`source_array = np.ascontiguousarray(`
`180`	`182`	`source_array, dtype="uint16" if dtype == "bfloat16" else dtype`
`181`	`183`	`)`