CMakeLists.txt 1.1 KB
Newer Older
T
tensor-tang 已提交
1 2 3

cc_library(jit_kernel_refer SRCS refer.cc DEPS jit_kernel_base)
set(JIT_KERNEL_DEPS ${JIT_KERNEL_DEPS} jit_kernel_refer PARENT_SCOPE)
T
tensor-tang 已提交
4 5 6 7 8 9

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

# use refer kernel by name
T
tensor-tang 已提交
10 11 12 13 14 15
USE_JITKERNEL_REFER(kVMul)
USE_JITKERNEL_REFER(kVAdd)
USE_JITKERNEL_REFER(kVAddRelu)
USE_JITKERNEL_REFER(kVSub)
USE_JITKERNEL_REFER(kVScal)
USE_JITKERNEL_REFER(kVAddBias)
16
USE_JITKERNEL_REFER(kVCopy)
T
tensor-tang 已提交
17 18 19 20 21 22 23 24 25 26 27 28 29
USE_JITKERNEL_REFER(kVRelu)
USE_JITKERNEL_REFER(kVIdentity)
USE_JITKERNEL_REFER(kVExp)
USE_JITKERNEL_REFER(kVSigmoid)
USE_JITKERNEL_REFER(kVTanh)
USE_JITKERNEL_REFER(kLSTMCtHt)
USE_JITKERNEL_REFER(kLSTMC1H1)
USE_JITKERNEL_REFER(kGRUH1)
USE_JITKERNEL_REFER(kGRUHtPart1)
USE_JITKERNEL_REFER(kGRUHtPart2)
USE_JITKERNEL_REFER(kCRFDecoding)
USE_JITKERNEL_REFER(kLayerNorm)
USE_JITKERNEL_REFER(kNCHW16CMulNC)
T
tensor-tang 已提交
30
USE_JITKERNEL_REFER(kSeqPool)
T
tensor-tang 已提交
31
USE_JITKERNEL_REFER(kMatMul)
T
tensor-tang 已提交
32
USE_JITKERNEL_REFER(kVSquare)
33 34 35
USE_JITKERNEL_REFER(kHSum)
USE_JITKERNEL_REFER(kHMax)
USE_JITKERNEL_REFER(kSoftmax)
36
USE_JITKERNEL_REFER(kEmbSeqPool)
37
USE_JITKERNEL_REFER(kSgd)
38
USE_JITKERNEL_REFER(kVBroadcast)