cc_library(jit_kernel_refer SRCS refer.cc DEPS jit_kernel_base) set(JIT_KERNEL_DEPS ${JIT_KERNEL_DEPS} jit_kernel_refer PARENT_SCOPE)
function(USE_JITKERNEL_REFER TARGET) file(APPEND ${jit_file} "USE_JITKERNEL_REFER(${TARGET});\n") endfunction() # use refer kernel by name
USE_JITKERNEL_REFER(kVMul) USE_JITKERNEL_REFER(kVAdd) USE_JITKERNEL_REFER(kVAddRelu) USE_JITKERNEL_REFER(kVSub) USE_JITKERNEL_REFER(kVScal) USE_JITKERNEL_REFER(kVAddBias) 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)
USE_JITKERNEL_REFER(kSeqPool)
USE_JITKERNEL_REFER(kMatMul)
