From a25cc9d153a1a84b0924362d485ff4f59d478da8 Mon Sep 17 00:00:00 2001 From: shademe Date: Wed, 7 Sep 2022 14:03:08 +0200 Subject: [PATCH] `util`: Explicitly call `__dlpack__` built-in method in `xp2tensorflow` `tf.experimental.dlpack.from_dlpack` expects a `PyCapsule` object. --- thinc/util.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/thinc/util.py b/thinc/util.py index 7add0aaf7..43fb115d2 100644 --- a/thinc/util.py +++ b/thinc/util.py @@ -404,7 +404,8 @@ def xp2tensorflow( dlpack_tensor = xp_tensor.toDlpack() # type: ignore tf_tensor = tf.experimental.dlpack.from_dlpack(dlpack_tensor) elif hasattr(xp_tensor, "__dlpack__"): - tf_tensor = tf.experimental.dlpack.from_dlpack(xp_tensor) + dlpack_tensor = xp_tensor.__dlpack__() # type: ignore + tf_tensor = tf.experimental.dlpack.from_dlpack(dlpack_tensor) else: tf_tensor = tf.convert_to_tensor(xp_tensor) if as_variable: