"...text-generation-inference.git" did not exist on "dfbd9a39a254ae7596bc3b53725d45a06e214a35"
Commit 6d5aba2f authored by rusty1s's avatar rusty1s
Browse files

print warn on single-threaded

parent c3026977
......@@ -25,8 +25,9 @@ def get_extensions():
info = parallel_info()
if 'parallel backend: OpenMP' in info and 'OpenMP not found' not in info:
print('Using OpenMP')
extra_compile_args['cxx'] += ['-DAT_PARALLEL_OPENMP', '-fopenmp']
else:
print('Compiling without OpenMP...')
if WITH_CUDA:
Extension = CUDAExtension
......
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