We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
libnuma
1 parent 8bf809e commit 1f6595dCopy full SHA for 1f6595d
skyrl-train/skyrl_train/workers/worker.py
@@ -155,7 +155,13 @@ class bitmask_t(Structure):
155
("maskp", POINTER(c_ulong)),
156
]
157
158
- LIBNUMA = CDLL(find_library("numa"))
+ try:
159
+ LIBNUMA = CDLL(find_library("numa"))
160
+ except Exception as e:
161
+ logger.error(f"Skipping NUMA affinity setup because libnuma is not installed: {e}")
162
+ _SET_AFFINITY = True
163
+ return
164
+
165
LIBNUMA.numa_parse_nodestring.argtypes = [c_char_p]
166
LIBNUMA.numa_parse_nodestring.restype = POINTER(bitmask_t)
167
LIBNUMA.numa_run_on_node_mask.argtypes = [POINTER(bitmask_t)]
0 commit comments