We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 0688eb7 commit 3632cbdCopy full SHA for 3632cbd
tests/fserver/test_fserver.py
@@ -1,10 +1,15 @@
1
-import torch, os
+import torch, os, sys
2
import time
3
+
4
+old_flags = sys.getdlopenflags()
5
+sys.setdlopenflags(sys.getdlopenflags() | 0x100)
6
import fserver_lib as f
7
+sys.setdlopenflags(old_flags)
8
9
is_worker = os.environ.get('DMLC_ROLE') == 'worker'
10
is_server = os.environ.get('DMLC_ROLE') == 'server'
11
-f.init()
12
+f.init("cmake_build/plugins/klx_backend/libklx_backend.so")
13
14
if is_worker:
15
gpu = os.environ.get('STEPMESH_GPU')
0 commit comments