gnh1201/wasm-micro-runtime
1
0
Fork 0
mirror of https://github.com/bytecodealliance/wasm-micro-runtime.git synced 2025-05-18 07:31:14 +00:00
Code Issues Actions 10 Packages Projects Releases Wiki Activity

Implement codegen lookupswitch (#1066)

Browse Source
This commit is contained in:
Wenyong Huang 2022-04-04 08:24:12 +08:00 committed by GitHub
parent 4d966d45ee
commit 883ce5d875
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 40 additions and 40 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

80
core/iwasm/fast-jit/cg/x86-64/jit_codegen_x86_64.cpp
View File

@ -3661,7 +3661,7 @@ fail:
} }
/* jmp to dst label */ /* jmp to dst label */
#define JMP_TO_LABEL(stream_offset, label_dst, label_src) \ #define JMP_TO_LABEL(label_dst, label_src) \
do { \ do { \
if (label_is_ahead(cc, label_dst, label_src)) { \ if (label_is_ahead(cc, label_dst, label_src)) { \
int32 _offset = label_offsets[label_dst] \ int32 _offset = label_offsets[label_dst] \
@ -3726,6 +3726,8 @@ fail:
* *
* @param cc the compiler context * @param cc the compiler context
* @param a the assembler to emit the code * @param a the assembler to emit the code
* @param jmp_info_list the jmp info list
* @param label_offsets the offsets of each label
* @param label_src the index of src label * @param label_src the index of src label
* @param key the entry key * @param key the entry key
* @param opnd the lookup switch operand * @param opnd the lookup switch operand
@ -3734,10 +3736,10 @@ fail:
* @return true if success, false if failed * @return true if success, false if failed
*/ */
static bool static bool
lookupswitch_imm(JitCompContext *cc, x86::Assembler &a, int32 label_src, lookupswitch_imm(JitCompContext *cc, x86::Assembler &a, bh_list *jmp_info_list,
int32 key, const JitOpndLookupSwitch *opnd, bool is_last_insn) uint32 *label_offsets, int32 label_src, int32 key,
const JitOpndLookupSwitch *opnd, bool is_last_insn)
{ {
#if 0
uint32 i; uint32 i;
int32 label_dst; int32 label_dst;
@ -3745,23 +3747,22 @@ lookupswitch_imm(JitCompContext *cc, x86::Assembler &a, int32 label_src,
if (key == opnd->match_pairs[i].value) { if (key == opnd->match_pairs[i].value) {
label_dst = jit_reg_no(opnd->match_pairs[i].target); label_dst = jit_reg_no(opnd->match_pairs[i].target);
if (!(is_last_insn if (!(is_last_insn
&& label_is_neighboring(cc, label_src, label_dst))) && label_is_neighboring(cc, label_src, label_dst))) {
JMP_TO_LABEL(stream_offset, label_dst, label_src); JMP_TO_LABEL(label_dst, label_src);
}
return true; return true;
} }
if (opnd->default_target) { if (opnd->default_target) {
label_dst = jit_reg_no(opnd->default_target); label_dst = jit_reg_no(opnd->default_target);
if (!(is_last_insn && label_is_neighboring(cc, label_src, label_dst))) if (!(is_last_insn && label_is_neighboring(cc, label_src, label_dst))) {
JMP_TO_LABEL(stream_offset, label_dst, label_src); JMP_TO_LABEL(label_dst, label_src);
}
} }
return true; return true;
fail: fail:
return false; return false;
#endif
return false;
} }
/** /**
@ -3769,6 +3770,8 @@ fail:
* *
* @param cc the compiler context * @param cc the compiler context
* @param a the assembler to emit the code * @param a the assembler to emit the code
* @param jmp_info_list the jmp info list
* @param label_offsets the offsets of each label
* @param label_src the index of src label * @param label_src the index of src label
* @param reg_no the no of entry register * @param reg_no the no of entry register
* @param opnd the lookup switch operand * @param opnd the lookup switch operand
@ -3777,47 +3780,44 @@ fail:
* @return true if success, false if failed * @return true if success, false if failed
*/ */
static bool static bool
lookupswitch_r(JitCompContext *cc, x86::Assembler &a, int32 label_src, lookupswitch_r(JitCompContext *cc, x86::Assembler &a, bh_list *jmp_info_list,
int32 reg_no, const JitOpndLookupSwitch *opnd, bool is_last_insn) uint32 *label_offsets, int32 label_src, int32 reg_no,
const JitOpndLookupSwitch *opnd, bool is_last_insn)
{ {
#if 0
JmpInfo *node; JmpInfo *node;
Imm_Opnd imm; Imm imm;
uint32 i, stream_offset_new; uint32 i;
int32 label_dst; int32 label_dst;
for (i = 0; i < opnd->match_pairs_num; i++) { for (i = 0; i < opnd->match_pairs_num; i++) {
imm_from_sz_v_s(imm, SZ32, opnd->match_pairs[i].value, true); imm.setValue(opnd->match_pairs[i].value);
alu_r_imm(cmp, regs_I32[reg_no], imm); a.cmp(regs_i32[reg_no], imm);
label_dst = jit_reg_no(opnd->match_pairs[i].target); label_dst = jit_reg_no(opnd->match_pairs[i].target);
imm_from_sz_v_s(imm, SZ32, label_dst, true); imm.setValue(label_dst);
node = jit_malloc(sizeof(JmpInfo)); node = (JmpInfo *)jit_malloc(sizeof(JmpInfo));
if (!node) if (!node)
GOTO_FAIL; GOTO_FAIL;
node->type = JMP_DST_LABEL; node->type = JMP_DST_LABEL;
node->label_src = label_src; node->label_src = label_src;
node->dst_info.label_dst = label_dst; node->dst_info.label_dst = label_dst;
node->offset = (int32)(stream + 2 - (*stream_ptr - stream_offset)); node->offset = a.code()->sectionById(0)->buffer().size() + 2;
bh_list_insert(jmp_info_list, node); bh_list_insert(jmp_info_list, node);
je(imm); a.je(imm);
} }
if (opnd->default_target) { if (opnd->default_target) {
label_dst = jit_reg_no(opnd->default_target); label_dst = jit_reg_no(opnd->default_target);
stream_offset_new = stream_offset + stream - *stream_ptr;
if (!(is_last_insn && label_is_neighboring(cc, label_src, label_dst))) if (!(is_last_insn && label_is_neighboring(cc, label_src, label_dst)))
JMP_TO_LABEL(stream_offset_new, label_dst, label_src); JMP_TO_LABEL(label_dst, label_src);
} }
return true; return true;
fail: fail:
return false; return false;
#endif
return false;
} }
/** /**
@ -3825,6 +3825,8 @@ fail:
* *
* @param cc the compiler context * @param cc the compiler context
* @param a the assembler to emit the code * @param a the assembler to emit the code
* @param jmp_info_list the jmp info list
* @param label_offsets the offsets of each label
* @param label_src the index of src label * @param label_src the index of src label
* @param opnd the lookup switch operand * @param opnd the lookup switch operand
* @param is_last_insn if current insn is the last insn of current block * @param is_last_insn if current insn is the last insn of current block
@ -3832,34 +3834,33 @@ fail:
* @return true if success, false if failed * @return true if success, false if failed
*/ */
static bool static bool
lower_lookupswitch(JitCompContext *cc, x86::Assembler &a, int32 label_src, lower_lookupswitch(JitCompContext *cc, x86::Assembler &a,
const JitOpndLookupSwitch *opnd, bool is_last_insn) bh_list *jmp_info_list, uint32 *label_offsets,
int32 label_src, const JitOpndLookupSwitch *opnd,
bool is_last_insn)
{ {
#if 0
JitReg r0 = opnd->value; JitReg r0 = opnd->value;
int32 key, reg_no; int32 key, reg_no;
CHECK_KIND(r0, JIT_REG_KIND_I32); CHECK_KIND(r0, JIT_REG_KIND_I32);
CHECK_KIND(opnd->default_target, JIT_REG_KIND_L4); CHECK_KIND(opnd->default_target, JIT_REG_KIND_L32);
if (jit_reg_is_const(r0)) { if (jit_reg_is_const(r0)) {
key = jit_cc_get_const_I32(cc, r0); key = jit_cc_get_const_I32(cc, r0);
if (!lookupswitch_imm(cc, &stream, stream_offset, label_src, key, opnd, if (!lookupswitch_imm(cc, a, jmp_info_list, label_offsets, label_src,
is_last_insn)) key, opnd, is_last_insn))
GOTO_FAIL; GOTO_FAIL;
} }
else { else {
reg_no = jit_reg_no(r0); reg_no = jit_reg_no(r0);
if (!lookupswitch_r(cc, &stream, stream_offset, label_src, reg_no, opnd, if (!lookupswitch_r(cc, a, jmp_info_list, label_offsets, label_src,
is_last_insn)) reg_no, opnd, is_last_insn))
GOTO_FAIL; GOTO_FAIL;
} }
return true; return true;
fail: fail:
return false; return false;
#endif
return false;
} }
/** /**
@ -4301,8 +4302,7 @@ jit_codegen_gen_native(JitCompContext *cc)
if (!(is_last_insn if (!(is_last_insn
&& label_is_neighboring(cc, label_index, && label_is_neighboring(cc, label_index,
jit_reg_no(r0)))) jit_reg_no(r0))))
JMP_TO_LABEL(stream_offset, jit_reg_no(r0), JMP_TO_LABEL(jit_reg_no(r0), label_index);
label_index);
break; break;
case JIT_OP_BEQ: case JIT_OP_BEQ:
@ -4326,8 +4326,8 @@ jit_codegen_gen_native(JitCompContext *cc)
case JIT_OP_LOOKUPSWITCH: case JIT_OP_LOOKUPSWITCH:
{ {
JitOpndLookupSwitch *opnd = jit_insn_opndls(insn); JitOpndLookupSwitch *opnd = jit_insn_opndls(insn);
if (!lower_lookupswitch(cc, a, label_index, opnd, if (!lower_lookupswitch(cc, a, jmp_info_list, label_offsets,
is_last_insn)) label_index, opnd, is_last_insn))
GOTO_FAIL; GOTO_FAIL;
break; break;
} }