update jit_uni_vcvtneps2bf16 arithmetic_mode for convert_truncation_e…

…mitter and store_emitter
openvinotoolkit · Nov 29, 2024 · d038a9a · d038a9a
1 parent f0b3f10
commit d038a9a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/src/plugins/intel_cpu/src/emitters/plugin/x64/jit_conversion_emitters.cpp b/src/plugins/intel_cpu/src/emitters/plugin/x64/jit_conversion_emitters.cpp
@@ -58,6 +58,9 @@ void jit_convert_emitter::float2bfloat(const std::vector<size_t> &in_vec_idxs, c
 jit_convert_truncation_emitter::jit_convert_truncation_emitter(jit_generator *host, cpu_isa_t host_isa,
                                                                const std::shared_ptr<ov::Node>& node, ov::element::Type exec_prc)
         : jit_convert_emitter(host, host_isa, node, exec_prc) {
+    if (uni_vcvtneps2bf16)
+        uni_vcvtneps2bf16.reset(new jit_uni_vcvtneps2bf16(host, host_isa, exec_prc, arithmetic_mode::truncation));
+
     prepare_table();
 }
 

diff --git a/src/plugins/intel_cpu/src/emitters/plugin/x64/jit_load_store_emitters.cpp b/src/plugins/intel_cpu/src/emitters/plugin/x64/jit_load_store_emitters.cpp
@@ -621,7 +621,7 @@ jit_store_emitter::jit_store_emitter(dnnl::impl::cpu::x64::jit_generator *host,
     prepare_table();
     v_len_elt_ = get_vec_length() / exec_prc.size();
     store_size_ = store_num * dst_prc.size();
-    uni_vcvtneps2bf16_.reset(new jit_uni_vcvtneps2bf16(host, host_isa));
+    uni_vcvtneps2bf16_.reset(new jit_uni_vcvtneps2bf16(host, host_isa, exec_prc, mode));
 }
 
 inline bool jit_store_emitter::is_saturation() const {