Skip to content

Commit

Permalink
fix a arm error
Browse files Browse the repository at this point in the history
  • Loading branch information
xczhai committed Nov 25, 2024
1 parent 688d26b commit aa4c1bb
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 3 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,8 @@ class LLMMLPNode : public ov::op::Op {
// 2: up_proj
// 3: down_proj
LLMMLPNode(const OutputVector& args, const Config& cfg, const ov::element::Type output_type = ov::element::undefined)
: Op(args), m_args(args), m_config(cfg), m_output_type(output_type) {
: Op(args), m_config(cfg), m_output_type(output_type) {
m_args = args;
validate_and_infer_types();
}

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -405,19 +405,20 @@ void Transformations::PreLpt(const std::vector<ov::element::Type>& defaultPrecis
precisions_map fp_convert_precision_map = {{ov::element::f32, ov::element::f16}};
#if defined(OPENVINO_ARCH_ARM) || defined(OPENVINO_ARCH_ARM64)
type_to_fuse_map fuse_map = {{ov::opset1::FakeQuantize::get_type_info_static(), fuse_type_to_fq}};
const bool store_original_precision_as_rt_attribute = false;
#else
type_to_fuse_map fuse_map = {{ov::op::PagedAttentionExtension::get_type_info_static(), fuse_type_to_pa}};
const bool store_original_precision_as_rt_attribute = true;
#endif
const bool keep_precision_sensitive_in_fp32 = true;
const bool need_convert_input_output_precision = false;
const bool save_original_precision_attribute = true;
CPU_REGISTER_PASS_COMMON(manager,
ov::pass::ConvertPrecision,
fp_convert_precision_map,
fuse_map,
keep_precision_sensitive_in_fp32,
need_convert_input_output_precision,
save_original_precision_attribute);
store_original_precision_as_rt_attribute);
}
CPU_REGISTER_PASS_COMMON(manager, ov::pass::KeepConstAndDecompression);
CPU_SET_CALLBACK_COMMON(manager,
Expand Down

0 comments on commit aa4c1bb

Please sign in to comment.