[llvm] 5375353 - TensorFlowCompile: Add object file to list of sources rather than LINK_LIBS

[Matthias Braun via llvm-commits]

Author: Matthias Braun
Date: 2022-06-01T09:04:48-07:00
New Revision: 53753531bc19bae816cf0cbf9e713ce0333ba091

URL: https://github.com/llvm/llvm-project/commit/53753531bc19bae816cf0cbf9e713ce0333ba091
DIFF: https://github.com/llvm/llvm-project/commit/53753531bc19bae816cf0cbf9e713ce0333ba091.diff

LOG: TensorFlowCompile: Add object file to list of sources rather than LINK_LIBS

Differential Revision: https://reviews.llvm.org/D126736

Added: 
    

Modified: 
    llvm/cmake/modules/TensorFlowCompile.cmake

Removed: 
    


################################################################################
diff  --git a/llvm/cmake/modules/TensorFlowCompile.cmake b/llvm/cmake/modules/TensorFlowCompile.cmake
index 861e559a3813a..80e361eba6895 100644
--- a/llvm/cmake/modules/TensorFlowCompile.cmake
+++ b/llvm/cmake/modules/TensorFlowCompile.cmake
@@ -110,8 +110,8 @@ function(tf_find_and_compile model default_url default_path test_model_generator
     tf_compile(${LLVM_ML_MODELS_ABSOLUTE} ${tag_set} ${signature_def_key} ${fname} ${cpp_class} ${hdr_file} ${obj_file})
   endif()
 
-  set(GeneratedMLSources ${GeneratedMLSources} ${GENERATED_HEADERS} PARENT_SCOPE)
+  set(GeneratedMLSources ${GeneratedMLSources} ${GENERATED_OBJS} ${GENERATED_HEADERS} PARENT_SCOPE)
   set(MLDeps ${MLDeps} tf_xla_runtime PARENT_SCOPE)
-  set(MLLinkDeps ${MLLinkDeps} tf_xla_runtime ${GENERATED_OBJS} PARENT_SCOPE)
+  set(MLLinkDeps ${MLLinkDeps} tf_xla_runtime PARENT_SCOPE)
   add_definitions(-DLLVM_HAVE_TF_AOT_${fname_allcaps})
 endfunction()