From e88bfde19b370a99753e14c71b4bfe5d073e5f25 Mon Sep 17 00:00:00 2001 From: Chao Hui Date: Sun, 18 Aug 2024 14:05:29 +0800 Subject: [PATCH] fix: cudf Buffer memory --- 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()