CMakeLists.txt 817 字节
Newer Older
T
tensor-tang 已提交
1

T
tensor-tang 已提交
2 3 4 5 6 7
# set(use_jit_file ${PADDLE_BINARY_DIR}/paddle/fluid/operators/jit/kernels.h)
# file(WRITE ${pass_file} "// Generated by the paddle/fluid/framework/ir/CMakeLists.txt.  DO NOT EDIT!\n\n")
# file(APPEND ${pass_file} "\#pragma once\n")
# file(APPEND ${pass_file} "\#include \"paddle/fluid/framework/ir/pass.h\"\n")


T
tensor-tang 已提交
8 9
set(JIT_KERNEL_DEPS cpu_info cblas gflags enforce place)

T
tensor-tang 已提交
10 11 12
file(GLOB jit_kernel_cc_srcs RELATIVE "${CMAKE_CURRENT_SOURCE_DIR}" "*.cc")
list(REMOVE_ITEM jit_kernel_cc_srcs jit_test.cc)
cc_library(jit_kernel_base SRCS ${jit_kernel_cc_srcs} DEPS ${JIT_KERNEL_DEPS})
T
tensor-tang 已提交
13 14

add_subdirectory(refer)
T
tensor-tang 已提交
15
add_subdirectory(more)
T
tensor-tang 已提交
16
if(WITH_XBYAK)
T
tensor-tang 已提交
17
    add_subdirectory(gen)
T
tensor-tang 已提交
18 19
endif()

T
tensor-tang 已提交
20 21
cc_library(jit_kernel_helper SRCS ${jit_kernel_cc_srcs} DEPS ${JIT_KERNEL_DEPS})
cc_test(jit_kernel_test SRCS test.cc DEPS jit_kernel_helper)