Skip to content

Commit c7dacb4

Browse files
committed
Fix issues
1 parent bf9232f commit c7dacb4

2 files changed

Lines changed: 2 additions & 1 deletion

File tree

plugin/interservice/flagcx_wrapper.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -500,6 +500,7 @@ def __init__(self, so_file: Optional[str] = None):
500500

501501
def __del__(self):
502502
# free flagcx device handle
503+
if hasattr(self, '_funcs') and hasattr(self, 'devHandle'):
503504
self.FLAGCX_CHECK(self._funcs["flagcxDeviceHandleFree"](self.devHandle))
504505

505506
def flagcxGetErrorString(self, result: flagcxResult_t) -> str:

plugin/interservice/test_triton_lsa.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -137,8 +137,8 @@ def _cleanup_flagcx_allocator_wrapper():
137137

138138

139139
import atexit
140-
atexit.register(_cleanup_flagcx_mem_pool)
141140
atexit.register(_cleanup_flagcx_allocator_wrapper)
141+
atexit.register(_cleanup_flagcx_mem_pool)
142142

143143

144144
# ============================================================

0 commit comments

Comments
 (0)