read copiler from cmakelists in setup.py.in (#31)

NervanaSystems · Nov 28, 2019 · 8a9ed91 · 8a9ed91
1 parent 33975b4
commit 8a9ed91
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/setup.py.in b/src/setup.py.in
@@ -56,7 +56,7 @@ if '${OPENFABRICS_FOUND}' == 'TRUE':
     lib_dirs += ' -L${OPENFABRICS_LIBRARY_DIRS} '
     libs += ' -lfabric '
 
-os.environ["CC"] = "g++"
+os.environ["CC"] = "${CMAKE_CXX_COMPILER}"
 src_list = "${SRC}".split(";")
 sources = [
     '${CMAKE_CURRENT_SOURCE_DIR}/'+src for src in src_list