fix

microsoft · Aug 3, 2022 · 702b132 · 702b132
1 parent 19ce1d6
commit 702b132
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/python/nnfusion/executor.py b/src/python/nnfusion/executor.py
@@ -233,7 +233,7 @@ def _maybe_reserve_mem(self, device):
         get_workspace_size = getattr(self.libnnf, 'get_workspace_size', None)
         if get_workspace_size is None:
             return None
-        get_workspace_size.restype = ctypes.c_int64
+        get_workspace_size.restype = ctypes.c_size_t
         n_byte = get_workspace_size()
         if not n_byte:
             return None