Skip to content

Commit 4689ead

Browse files
committed
Add note to README
1 parent 90c1ae5 commit 4689ead

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

README.md

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1557,6 +1557,10 @@ input0 = pb_utils.Tensor.from_dlpack("INPUT0", pytorch_tensor)
15571557
This method only supports contiguous Tensors that are in C-order. If the tensor
15581558
is not C-order contiguous an exception will be raised.
15591559

1560+
For python models with input or output tensors of type BFloat16 (BF16), the
1561+
`as_numpy()` method is not supported, and the `from_dlpack` and `to_dlpack`
1562+
methods must be used instead.
1563+
15601564
## `pb_utils.Tensor.is_cpu() -> bool`
15611565

15621566
This function can be used to check whether a tensor is placed in CPU or not.

0 commit comments

Comments
 (0)