Skip to content

Commit

Permalink
review comments
Browse files Browse the repository at this point in the history
  • Loading branch information
eshoguli committed Mar 21, 2024
1 parent 2b3ad65 commit cf67aa7
Showing 1 changed file with 3 additions and 3 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -516,7 +516,7 @@ size_t jit_select_emitter::get_aux_vecs_count() const { return 1; }
size_t jit_select_emitter::get_aux_gprs_count() const { return 1; }

std::set<std::vector<element::Type>> jit_select_emitter::get_supported_precisions(const std::shared_ptr<ov::Node>& node) {
return {{element::f32, element::f32}};
return {{element::f32, element::f32, element::f32}};
}

void jit_select_emitter::emit_impl(const std::vector<size_t>& in_vec_idxs, const std::vector<size_t>& out_vec_idxs) const {
Expand All @@ -538,8 +538,8 @@ void jit_select_emitter::emit_isa(const std::vector<size_t> &in_vec_idxs, const
const TReg dst = TReg(out_vec_idxs[0]);
const TReg aux = TReg(aux_vec_idxs[0]);

h->ld1r(aux.s, table_val2("one"));
h->facge(aux.s, src1.s, aux.s);
h->eor(aux.b16, aux.b16, aux.b16);
h->fcmgt(aux.s, src1.s, aux.s);

h->bsl(aux.b16, src2.b16, src3.b16);
h->mov(dst.b16, aux.b16);
Expand Down

0 comments on commit cf67aa7

Please sign in to comment.