Skip to content

Commit 3632cbd

Browse files
committed
[DONTMERGE] test for plugin
1 parent 0688eb7 commit 3632cbd

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

tests/fserver/test_fserver.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,15 @@
1-
import torch, os
1+
import torch, os, sys
22
import time
3+
4+
old_flags = sys.getdlopenflags()
5+
sys.setdlopenflags(sys.getdlopenflags() | 0x100)
36
import fserver_lib as f
7+
sys.setdlopenflags(old_flags)
8+
49
is_worker = os.environ.get('DMLC_ROLE') == 'worker'
510
is_server = os.environ.get('DMLC_ROLE') == 'server'
611

7-
f.init()
12+
f.init("cmake_build/plugins/klx_backend/libklx_backend.so")
813

914
if is_worker:
1015
gpu = os.environ.get('STEPMESH_GPU')

0 commit comments

Comments
 (0)