| @@ -22,6 +22,10 @@ | |||||
| namespace mindspore { | namespace mindspore { | ||||
| namespace kernel { | namespace kernel { | ||||
| void LstmCPUKernel::InitKernel(const CNodePtr &kernel_node) { | void LstmCPUKernel::InitKernel(const CNodePtr &kernel_node) { | ||||
| #ifdef PLATFORM_86 | |||||
| _MM_SET_FLUSH_ZERO_MODE(_MM_FLUSH_ZERO_ON); | |||||
| _MM_SET_DENORMALS_ZERO_MODE(_MM_DENORMALS_ZERO_ON); | |||||
| #endif | |||||
| MS_EXCEPTION_IF_NULL(kernel_node); | MS_EXCEPTION_IF_NULL(kernel_node); | ||||
| using tag = dnnl::memory::format_tag; | using tag = dnnl::memory::format_tag; | ||||
| using dim = dnnl::memory::dims; | using dim = dnnl::memory::dims; | ||||
| @@ -16,6 +16,12 @@ | |||||
| #ifndef MINDSPORE_CCSRC_KERNEL_CPU_LSTM_CPU_KERNEL_H_ | #ifndef MINDSPORE_CCSRC_KERNEL_CPU_LSTM_CPU_KERNEL_H_ | ||||
| #define MINDSPORE_CCSRC_KERNEL_CPU_LSTM_CPU_KERNEL_H_ | #define MINDSPORE_CCSRC_KERNEL_CPU_LSTM_CPU_KERNEL_H_ | ||||
| #if defined(__x86_64__) || defined(__amd64__) || defined(_M_IX86) || defined(_M_X64) | |||||
| #define PLATFORM_86 | |||||
| #endif | |||||
| #ifdef PLATFORM_86 | |||||
| #include <pmmintrin.h> | |||||
| #endif | |||||
| #include <vector> | #include <vector> | ||||
| #include <memory> | #include <memory> | ||||
| #include "kernel/cpu/mkldnn/mkl_cpu_kernel.h" | #include "kernel/cpu/mkldnn/mkl_cpu_kernel.h" | ||||