#ifdef PYBIND_AVX_MKLML // Generated by the paddle/fluid/operators/jit/CMakeLists.txt. DO NOT EDIT! #pragma once #include "paddle/fluid/operators/jit/helper.h" #include "paddle/fluid/operators/jit/registry.h" USE_JITKERNEL_REFER(kVMul); USE_JITKERNEL_REFER(kVAdd); USE_JITKERNEL_REFER(kVAddRelu); USE_JITKERNEL_REFER(kVSub); USE_JITKERNEL_REFER(kVScal); USE_JITKERNEL_REFER(kStrideScal); USE_JITKERNEL_REFER(kVAddBias); USE_JITKERNEL_REFER(kVCopy); 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); USE_JITKERNEL_REFER(kVSquare); USE_JITKERNEL_REFER(kHSum); USE_JITKERNEL_REFER(kHMax); USE_JITKERNEL_REFER(kStrideASum); USE_JITKERNEL_REFER(kSoftmax); USE_JITKERNEL_REFER(kEmbSeqPool); USE_JITKERNEL_REFER(kSgd); USE_JITKERNEL_REFER(kVBroadcast); USE_JITKERNEL_MORE(kMatMul, mkl); USE_JITKERNEL_MORE(kVMul, mkl); USE_JITKERNEL_MORE(kVAdd, mkl); USE_JITKERNEL_MORE(kVScal, mkl); USE_JITKERNEL_MORE(kStrideScal, mkl); USE_JITKERNEL_MORE(kVExp, mkl); USE_JITKERNEL_MORE(kVSquare, mkl); USE_JITKERNEL_MORE(kVCopy, mkl); USE_JITKERNEL_MORE(kVSigmoid, mkl); USE_JITKERNEL_MORE(kVTanh, mkl); USE_JITKERNEL_MORE(kSeqPool, mkl); USE_JITKERNEL_MORE(kSoftmax, mkl); USE_JITKERNEL_MORE(kEmbSeqPool, mkl); USE_JITKERNEL_MORE(kSgd, mkl); USE_JITKERNEL_MORE(kVBroadcast, mkl); USE_JITKERNEL_MORE(kCRFDecoding, intrinsic); USE_JITKERNEL_MORE(kLayerNorm, intrinsic); USE_JITKERNEL_MORE(kVSigmoid, mix); USE_JITKERNEL_MORE(kVTanh, mix); USE_JITKERNEL_MORE(kLSTMCtHt, mix); USE_JITKERNEL_MORE(kLSTMC1H1, mix); USE_JITKERNEL_MORE(kGRUH1, mix); USE_JITKERNEL_MORE(kGRUHtPart1, mix); USE_JITKERNEL_MORE(kGRUHtPart2, mix); USE_JITKERNEL_MORE(kSoftmax, mix); USE_JITKERNEL_GEN(kMatMul); USE_JITKERNEL_GEN(kVMul); USE_JITKERNEL_GEN(kVAdd); USE_JITKERNEL_GEN(kVSub); USE_JITKERNEL_GEN(kVAddRelu); USE_JITKERNEL_GEN(kVScal); USE_JITKERNEL_GEN(kVAddBias); USE_JITKERNEL_GEN(kVRelu); USE_JITKERNEL_GEN(kVSquare); 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); USE_JITKERNEL_GEN(kSeqPool); USE_JITKERNEL_GEN(kHMax); USE_JITKERNEL_GEN(kHSum); USE_JITKERNEL_GEN(kEmbSeqPool); USE_JITKERNEL_GEN(kSgd); USE_JITKERNEL_GEN(kVBroadcast); #elif defined PYBIND_NOAVX_OPENBLAS // Generated by the paddle/fluid/operators/jit/CMakeLists.txt. DO NOT EDIT! #pragma once #include "paddle/fluid/operators/jit/helper.h" #include "paddle/fluid/operators/jit/registry.h" USE_JITKERNEL_REFER(kVMul); USE_JITKERNEL_REFER(kVAdd); USE_JITKERNEL_REFER(kVAddRelu); USE_JITKERNEL_REFER(kVSub); USE_JITKERNEL_REFER(kVScal); USE_JITKERNEL_REFER(kStrideScal); USE_JITKERNEL_REFER(kVAddBias); USE_JITKERNEL_REFER(kVCopy); 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); USE_JITKERNEL_REFER(kVSquare); USE_JITKERNEL_REFER(kHSum); USE_JITKERNEL_REFER(kHMax); USE_JITKERNEL_REFER(kStrideASum); USE_JITKERNEL_REFER(kSoftmax); USE_JITKERNEL_REFER(kEmbSeqPool); USE_JITKERNEL_REFER(kSgd); USE_JITKERNEL_REFER(kVBroadcast); USE_JITKERNEL_MORE(kVSigmoid, mix); USE_JITKERNEL_MORE(kVTanh, mix); USE_JITKERNEL_MORE(kLSTMCtHt, mix); USE_JITKERNEL_MORE(kLSTMC1H1, mix); USE_JITKERNEL_MORE(kGRUH1, mix); USE_JITKERNEL_MORE(kGRUHtPart1, mix); USE_JITKERNEL_MORE(kGRUHtPart2, mix); USE_JITKERNEL_MORE(kSoftmax, mix); USE_JITKERNEL_GEN(kMatMul); USE_JITKERNEL_GEN(kVMul); USE_JITKERNEL_GEN(kVAdd); USE_JITKERNEL_GEN(kVSub); USE_JITKERNEL_GEN(kVAddRelu); USE_JITKERNEL_GEN(kVScal); USE_JITKERNEL_GEN(kVAddBias); USE_JITKERNEL_GEN(kVRelu); USE_JITKERNEL_GEN(kVSquare); 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); USE_JITKERNEL_GEN(kSeqPool); USE_JITKERNEL_GEN(kHMax); USE_JITKERNEL_GEN(kHSum); USE_JITKERNEL_GEN(kEmbSeqPool); USE_JITKERNEL_GEN(kSgd); USE_JITKERNEL_GEN(kVBroadcast); #endif