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)
set(JIT_KERNEL_DEPS ${JIT_KERNEL_DEPS} xbyak jit_kernel_jitcode PARENT_SCOPE)
function(USE_JITKERNEL_GEN TARGET) file(APPEND ${jit_file} "USE_JITKERNEL_GEN(${TARGET});\n") endfunction() # use gen jitcode kernel by name
USE_JITKERNEL_GEN(kVMul) USE_JITKERNEL_GEN(kVAdd) #USE_JITKERNEL_GEN(kVSub) # TODO(TJ): enable me USE_JITKERNEL_GEN(kVAddRelu) USE_JITKERNEL_GEN(kVScal) USE_JITKERNEL_GEN(kVAddBias) USE_JITKERNEL_GEN(kVRelu) 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)
