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

T
tensor-tang 已提交
2 3 4
file(GLOB jitcode_cc_srcs RELATIVE "${CMAKE_CURRENT_SOURCE_DIR}" "*.cc")

cc_library(jit_kernel_jitcode SRCS ${jitcode_cc_srcs} DEPS jit_kernel_base xbyak)
T
tensor-tang 已提交
5
set(JIT_KERNEL_DEPS ${JIT_KERNEL_DEPS} xbyak jit_kernel_jitcode PARENT_SCOPE)
T
tensor-tang 已提交
6 7 8 9 10 11

function(USE_JITKERNEL_GEN TARGET)
    file(APPEND ${jit_file} "USE_JITKERNEL_GEN(${TARGET});\n")
endfunction()

# use gen jitcode kernel by name
T
tensor-tang 已提交
12 13
USE_JITKERNEL_GEN(kVMul)
USE_JITKERNEL_GEN(kVAdd)
T
tensor-tang 已提交
14
USE_JITKERNEL_GEN(kVSub)
T
tensor-tang 已提交
15 16 17 18
USE_JITKERNEL_GEN(kVAddRelu)
USE_JITKERNEL_GEN(kVScal)
USE_JITKERNEL_GEN(kVAddBias)
USE_JITKERNEL_GEN(kVRelu)
T
tensor-tang 已提交
19
USE_JITKERNEL_GEN(kVSquare)
T
tensor-tang 已提交
20 21 22 23 24 25 26 27 28 29
USE_JITKERNEL_GEN(kVIdentity)
USE_JITKERNEL_GEN(kVExp)
USE_JITKERNEL_GEN(kVSigmoid)
USE_JITKERNEL_GEN(kVTanh)
USE_JITKERNEL_GEN(kLSTMCtHt)
USE_JITKERNEL_GEN(kLSTMC1H1)
USE_JITKERNEL_GEN(kGRUH1)
USE_JITKERNEL_GEN(kGRUHtPart1)
USE_JITKERNEL_GEN(kGRUHtPart2)
USE_JITKERNEL_GEN(kNCHW16CMulNC)
T
tensor-tang 已提交
30
USE_JITKERNEL_GEN(kSeqPool)