Merge pull request #1703 from ShacharKagan/mem_fixes

pyverbs: Fix memory leaks and update Python/Cython compatibility

Author: rleon

pyverbs/base.pyx

@@ -36,12 +36,11 @@ cdef close_weakrefs(iterables):
     for it in iterables:
         if it is None:
             continue
-        while True:
-            try:
-                tmp = it.pop()
-                tmp.close()
-            except KeyError: # popping an empty set
-                break
+        items = list(it)
+        it.clear()
+        for item in items:
+            if item is not None:
+                item.close()
 
 
 cdef class PyverbsObject(object):

pyverbs/providers/mlx5/mlx5dv.pyx

@@ -276,7 +276,7 @@ cdef class Mlx5DevxObj(PyverbsCM):
 
     @property
     def obj(self):
-        return <object>self.obj
+        return <uintptr_t>self.obj
 
     def __dealloc__(self):
         self.close()

tests/test_mlx5_crypto.py

@@ -64,8 +64,8 @@ def check_crypto_caps(dev_name, is_wrapped_dek_mode, multi_block_support=False):
     :param multi_block_support: If True, check for multi-block support.
     """
     mlx5dv_attr = Mlx5DVContextAttr()
-    ctx = Mlx5Context(mlx5dv_attr, name=dev_name)
-    crypto_caps = ctx.query_mlx5_device().crypto_caps
+    with Mlx5Context(mlx5dv_attr, name=dev_name) as ctx:
+        crypto_caps = ctx.query_mlx5_device().crypto_caps
     failed_selftests = crypto_caps['failed_selftests']
     if failed_selftests:
         raise unittest.SkipTest(f'The device crypto selftest failed ({failed_selftests})')