File tree Expand file tree Collapse file tree 1 file changed +1
-5
lines changed
Expand file tree Collapse file tree 1 file changed +1
-5
lines changed Original file line number Diff line number Diff line change @@ -287,9 +287,6 @@ def test_mempool():
287287 buffer .close ()
288288
289289
290- import sys # noqa: E402
291-
292-
293290def 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 )
You can’t perform that action at this time.
0 commit comments