Skip to content

GitLab

"cacheflow/git@developer.sourcefind.cn:norm/vllm.git" did not exist on "aedba6d5ec4e1eaad10745f71970c10b601f9dc1"
Commit fa6872cd authored by Jiezhong Qiu's avatar Jiezhong Qiu
Browse files

Merge remote-tracking branch 'origin/master' into init_expert

parents da4a8e5e 02ead75d
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 1 deletion
setup.py
View file @ fa6872cd
...@@ -7,8 +7,10 @@ CUDA_HELPER = os.environ.get('CUDA_HELPER', '/usr/local/cuda/samples/common/inc' ...@@ -7,8 +7,10 @@ CUDA_HELPER = os.environ.get('CUDA_HELPER', '/usr/local/cuda/samples/common/inc'
cxx_flags = [ cxx_flags = [
'-I{}'.format(CUDA_HELPER) '-I{}'.format(CUDA_HELPER)
] ]
ext_libs = []
if os.environ.get('USE_NCCL', '0') == '1': if os.environ.get('USE_NCCL', '0') == '1':
cxx_flags.append('-DMOE_USE_NCCL') cxx_flags.append('-DMOE_USE_NCCL')
ext_libs.append('nccl')
if __name__ == '__main__': if __name__ == '__main__':
...@@ -34,7 +36,8 @@ if __name__ == '__main__': ...@@ -34,7 +36,8 @@ if __name__ == '__main__':
extra_compile_args={ extra_compile_args={
'cxx': cxx_flags, 'cxx': cxx_flags,
'nvcc': cxx_flags 'nvcc': cxx_flags
} },
libraries=ext_libs
) )
], ],
cmdclass={ cmdclass={
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment