Skip to content
Snippets Groups Projects
Commit 7def7d13 authored by Stephan Seitz's avatar Stephan Seitz
Browse files

Make use of p_tqdm==1.3

parent e457402d
No related branches found
No related tags found
No related merge requests found
Pipeline #20784 failed
......@@ -36,7 +36,7 @@ install_requires =
sympy>=0.5 # lower sympy version will have problems with a bug in floor when using interpolation
jinja2
tqdm
p_tqdm # for parallel compilation in tensorflow_jit
p_tqdm>=1.3 # for parallel compilation in tensorflow_jit
stringcase # for converting to camelcase for tensorflow
# The usage of test_requires is discouraged, see `Dependency Management` docs
test_requires =
......
......@@ -198,7 +198,6 @@ def compile_sources_and_load(host_sources,
object_files = []
sources = host_sources + cuda_sources
print('Compiling Tensorflow module...')
def compile(source):
is_cuda = source in cuda_sources
......@@ -221,7 +220,7 @@ def compile_sources_and_load(host_sources,
return object_file
# p_tqdm is just a parallel tqdm
object_files = p_tqdm.p_umap(compile, sources)
object_files = p_tqdm.p_umap(compile, sources, desc='Compiling Tensorflow module')
print('Linking Tensorflow module...')
module_file = link(object_files,
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment