build: fix ucx build (#330)

do not build ucx by default. Signed-off-by: Yulu Jia <[email protected]>
bytedance · Dec 6, 2020 · ff7f5dd · ff7f5dd
1 parent 38431f0
commit ff7f5dd
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/pre_setup.py b/pre_setup.py
@@ -5,3 +5,5 @@ def setup():
 
 def extra_make_option():
     return ""
+
+ucx_path = ""
diff --git a/setup.py b/setup.py
@@ -237,7 +237,7 @@ def has_rdma_header():
     return ret_code==0
 
 def build_ucx():
-    byteps_with_ucx = int(os.environ.get('BYTEPS_WITH_UCX', 1))
+    byteps_with_ucx = int(os.environ.get('BYTEPS_WITH_UCX', 0))
     return byteps_with_ucx
 
 def get_common_options(build_ext):
@@ -918,7 +918,7 @@ def build_extensions(self):
                 make_option += "-j "
             if has_rdma_header():
                 make_option += "USE_RDMA=1 "
-            if build_ucx:
+            if build_ucx():
                 make_option += 'USE_UCX=1 '
 
             make_option += pre_setup.extra_make_option()
Original file line number	Diff line number	Diff line change
Expand Up		@@ -5,3 +5,5 @@ def setup():

		def extra_make_option():
		return ""

		ucx_path = ""