forked from OSchip/llvm-project
Remove redundant environment variable XLA_FLAGS.
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:
parent
035217ff51
commit
6c848c28c2
|
@ -32,7 +32,8 @@ function(tfcompile model tag_set signature_def_key fname cpp_class)
|
|||
tfgetmodel(${model} LLVM_ML_MODELS_ABSOLUTE)
|
||||
message("Using model at " ${LLVM_ML_MODELS_ABSOLUTE})
|
||||
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}
|
||||
--tag_set ${tag_set}
|
||||
--signature_def_key ${signature_def_key}
|
||||
|
|
Loading…
Reference in New Issue