mirror of
https://github.com/RPCS3/llvm-mirror.git
synced 2024-11-23 11:13:28 +01:00
ddbad0ce22
Summary: This implementation uses a pre-trained model which is statically compiled into a native function. RFC: http://lists.llvm.org/pipermail/llvm-dev/2020-April/140763.html Reviewers: davidxl, jdoerfert, dblaikie Subscribers: mgorny, eraman, hiraditya, arphaman, llvm-commits Tags: #llvm Differential Revision: https://reviews.llvm.org/D81515
39 lines
1.5 KiB
CMake
39 lines
1.5 KiB
CMake
# Run the tensorflow compiler (saved_model_cli) on the saved model in the
|
|
# ${model} directory, looking for the ${tag_set} tag set, and the SignatureDef
|
|
# ${signature_def_key}.
|
|
# Produce a pair of files called ${fname}.h and ${fname}.o in the
|
|
# ${CMAKE_CURRENT_BINARY_DIR}. The generated header will define a C++ class
|
|
# called ${cpp_class} - which may be a namespace-qualified class name.
|
|
function(tfcompile model tag_set signature_def_key fname cpp_class)
|
|
if (IS_ABSOLUTE ${model})
|
|
set(LLVM_ML_MODELS_ABSOLUTE ${model})
|
|
else()
|
|
set(LLVM_ML_MODELS_ABSOLUTE
|
|
${CMAKE_CURRENT_SOURCE_DIR}/${model})
|
|
endif()
|
|
|
|
set(prefix ${CMAKE_CURRENT_BINARY_DIR}/${fname})
|
|
set(obj_file ${prefix}.o)
|
|
set(hdr_file ${prefix}.h)
|
|
add_custom_command(OUTPUT ${obj_file} ${hdr_file}
|
|
COMMAND "XLA_FLAGS=\"--xla_cpu_multi_thread_eigen=false\"" ${TENSORFLOW_AOT_COMPILER} aot_compile_cpu
|
|
--dir ${LLVM_ML_MODELS_ABSOLUTE}
|
|
--tag_set ${tag_set}
|
|
--signature_def_key ${signature_def_key}
|
|
--output_prefix ${prefix}
|
|
--cpp_class ${cpp_class}
|
|
--target_triple ${LLVM_HOST_TRIPLE}
|
|
)
|
|
|
|
# Aggregate the objects so that results of different tfcompile calls may be
|
|
# grouped into one target.
|
|
set(GENERATED_OBJS ${GENERATED_OBJS} ${obj_file} PARENT_SCOPE)
|
|
set_source_files_properties(${obj_file} PROPERTIES
|
|
GENERATED 1 EXTERNAL_OBJECT 1)
|
|
|
|
set(GENERATED_HEADERS ${GENERATED_HEADERS} ${hdr_file} PARENT_SCOPE)
|
|
set_source_files_properties(${hdr_file} PROPERTIES
|
|
GENERATED 1)
|
|
|
|
endfunction()
|