Fix build warning/error of jax-0.3.25-foss-2022a

easybuilders · Sep 12, 2024 · edcc8e4 · edcc8e4
1 parent 9de7a6b
commit edcc8e4
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/easybuild/easyconfigs/j/jax/jax-0.3.25-foss-2022a.eb b/easybuild/easyconfigs/j/jax/jax-0.3.25-foss-2022a.eb
@@ -35,6 +35,8 @@ local_tf_builddir = '%(builddir)s/' + local_tf_dir
 # replace remote TensorFlow repository with the local one from EB
 local_jax_prebuildopts = "sed -i -f jaxlib_local-tensorflow-repo.sed WORKSPACE && "
 local_jax_prebuildopts += "sed -i 's|EB_TF_REPOPATH|%s|' WORKSPACE && " % local_tf_builddir
+# Avoid warning (treated as error) in upb/table.c
+local_jaxlib_buildopts = '--bazel_options="--copt=-Wno-maybe-uninitialized" '
 
 use_pip = True
 
@@ -81,6 +83,7 @@ components = [
         ],
         'start_dir': 'jax-jaxlib-v%(version)s',
         'prebuildopts': local_jax_prebuildopts,
+        'buildopts': local_jaxlib_buildopts,
     }),
 ]