Skip to content

Commit 1fc2660

Browse files
add ipc buffer class
1 parent 5086134 commit 1fc2660

File tree

1 file changed

+1
-5
lines changed

1 file changed

+1
-5
lines changed

cuda_core/tests/test_memory.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -287,9 +287,6 @@ def test_mempool():
287287
buffer.close()
288288

289289

290-
import sys # noqa: E402
291-
292-
293290
def mempool_child_process(importer, queue):
294291
try:
295292
device = Device()
@@ -320,8 +317,7 @@ def mempool_child_process(importer, queue):
320317
# Verify the data matches expected pattern
321318
verify_ptr = ctypes.cast(int(verify_buffer.handle), ctypes.POINTER(ctypes.c_byte))
322319
for i in range(64):
323-
sys.stderr.write(f"[{verify_ptr[i]}, {ctypes.c_byte(i)}] ")
324-
assert verify_ptr[i] == i, f"Data mismatch at index {i}"
320+
assert ctypes.c_byte(verify_ptr[i]).value == ctypes.c_byte(i).value, f"Data mismatch at index {i}"
325321

326322
# Write new pattern to the buffer using unified memory
327323
src_buffer = unified_mr.allocate(64)

0 commit comments

Comments
 (0)