From c4e49319d52191a606711506cf9b7899d7a84387 Mon Sep 17 00:00:00 2001 From: Hui Chao Date: Sun, 18 Aug 2024 22:47:43 +0800 Subject: [PATCH] ENH: support cudf Buffer BufferOwner structure (#798) Co-authored-by: mergify[bot] <37929162+mergify[bot]@users.noreply.github.com> --- python/xorbits/_mars/_utils.pyx | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/xorbits/_mars/_utils.pyx b/python/xorbits/_mars/_utils.pyx index 878d19e01..181238a23 100644 --- a/python/xorbits/_mars/_utils.pyx +++ b/python/xorbits/_mars/_utils.pyx @@ -286,7 +286,7 @@ def tokenize_cupy(ob): def tokenize_cudf(ob): from xoscar.serialization import serialize header, buffers = serialize(ob) - return iterative_tokenize([header] + [(buf.ptr, buf.size) for buf in buffers]) + return iterative_tokenize([header] + [(buf._owner._ptr, buf.size) for buf in buffers]) cdef Tokenizer tokenize_handler = Tokenizer()