py/asmx64: Support full range of regs in asm_x64_lea_disp_to_r64.

dpgeorge · dpgeorge · commit 8e1db993cd6b · 2022-05-17T14:25:51.000+10:00
Signed-off-by: Damien George &lt;damien@micropython.org&gt;
diff --git a/py/asmx64.c b/py/asmx64.c
@@ -319,9 +319,7 @@ void asm_x64_mov_mem64_to_r64(asm_x64_t *as, int src_r64, int src_disp, int dest
 
 STATIC void asm_x64_lea_disp_to_r64(asm_x64_t *as, int src_r64, int src_disp, int dest_r64) {
     // use REX prefix for 64 bit operation
-    assert(src_r64 < 8);
-    assert(dest_r64 < 8);
-    asm_x64_write_byte_2(as, REX_PREFIX | REX_W, OPCODE_LEA_MEM_TO_R64);
+    asm_x64_write_byte_2(as, REX_PREFIX | REX_W | REX_R_FROM_R64(dest_r64) | REX_B_FROM_R64(src_r64), OPCODE_LEA_MEM_TO_R64);
     asm_x64_write_r64_disp(as, dest_r64, src_r64, src_disp);
 }
 

Original file line number	Diff line number	Diff line change
`@@ -319,9 +319,7 @@ void asm_x64_mov_mem64_to_r64(asm_x64_t *as, int src_r64, int src_disp, int dest`
`319`	`319`
`320`	`320`	`STATIC void asm_x64_lea_disp_to_r64(asm_x64_t *as, int src_r64, int src_disp, int dest_r64) {`
`321`	`321`	`// use REX prefix for 64 bit operation`
`322`		`- assert(src_r64 < 8);`
`323`		`- assert(dest_r64 < 8);`
`324`		`- asm_x64_write_byte_2(as, REX_PREFIX \| REX_W, OPCODE_LEA_MEM_TO_R64);`
	`322`	`+ asm_x64_write_byte_2(as, REX_PREFIX \| REX_W \| REX_R_FROM_R64(dest_r64) \| REX_B_FROM_R64(src_r64), OPCODE_LEA_MEM_TO_R64);`
`325`	`323`	`asm_x64_write_r64_disp(as, dest_r64, src_r64, src_disp);`
`326`	`324`	`}`
`327`	`325`