CMakeLists.txt 676 字节
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
cc_library(jit_kernel_base SRCS kernels.cc jitcode_base.cc DEPS ${JIT_KERNEL_DEPS})
T
tensor-tang 已提交
11 12

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

cc_library(jit_kernel SRCS kernels.cc DEPS ${JIT_KERNEL_DEPS})
cc_test(jit_kernel_test SRCS test.cc DEPS jit_kernel)