Skip to content

Commit 741f219

Browse files
add ipc buffer class
1 parent 3577006 commit 741f219

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

cuda_core/tests/test_memory.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -403,7 +403,10 @@ def test_ipc_mempool():
403403

404404
verify_ptr = ctypes.cast(int(verify_buffer.handle), ctypes.POINTER(ctypes.c_byte))
405405
for i in range(64):
406-
assert verify_ptr[i] == 255 - i, f"Child process data not reflected in parent at index {i}"
406+
print(f" {verify_ptr[i]}, {ctypes.c_byte(255 - i)}")
407+
assert ctypes.c_byte(verify_ptr[i]) == ctypes.c_byte(
408+
255 - i
409+
), f"Child process data not reflected in parent at index {i}"
407410

408411
verify_buffer.close()
409412
buffer.close()

0 commit comments

Comments
 (0)