RPCS3/llvm-mirror
1
0
Fork 0
mirror of https://github.com/RPCS3/llvm-mirror.git synced 2024-11-25 12:12:47 +01:00
Code Issues Projects Releases Wiki Activity

Remove redundant environment variable XLA_FLAGS.

Browse Source 
If the flag is not set, the script saved_model_aot_compile.py in tensorflow will
default it to the correct value. However, in TF 2.5, the way the value is set in
TensorFlowCompile.cmake file triggers a build error.

Reviewed By: mtrofin

Differential Revision: https://reviews.llvm.org/D103972
This commit is contained in:
Jacob Hegna 2021-06-14 23:21:39 +00:00
parent dfb1691713
commit aeb9af0756
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
cmake/modules/TensorFlowCompile.cmake
View File

@ -32,7 +32,8 @@ function(tfcompile model tag_set signature_def_key fname cpp_class)
tfgetmodel(${model} LLVM_ML_MODELS_ABSOLUTE) tfgetmodel(${model} LLVM_ML_MODELS_ABSOLUTE)
message("Using model at " ${LLVM_ML_MODELS_ABSOLUTE}) message("Using model at " ${LLVM_ML_MODELS_ABSOLUTE})
add_custom_command(OUTPUT ${obj_file} ${hdr_file} add_custom_command(OUTPUT ${obj_file} ${hdr_file}
COMMAND "XLA_FLAGS=\"--xla_cpu_multi_thread_eigen=false\"" ${TENSORFLOW_AOT_COMPILER} aot_compile_cpu COMMAND ${TENSORFLOW_AOT_COMPILER} aot_compile_cpu
--multithreading false
--dir ${LLVM_ML_MODELS_ABSOLUTE} --dir ${LLVM_ML_MODELS_ABSOLUTE}
--tag_set ${tag_set} --tag_set ${tag_set}
--signature_def_key ${signature_def_key} --signature_def_key ${signature_def_key}