59 lines
1.4 KiB
Python
59 lines
1.4 KiB
Python
from optimizers.optimizer import Optimizer, PeepholeRule, Instruction
|
|
|
|
class X86_64Optimizer(Optimizer):
|
|
def __init__(self, lines, window_size = 5):
|
|
super().__init__(lines, window_size)
|
|
self.rules = [
|
|
PeepholeRule(
|
|
match=[
|
|
Instruction("setg",["al"]),
|
|
Instruction("movzx", ["rax", "al"]),
|
|
Instruction("push", ["rax"]),
|
|
Instruction("mov", ["eax", "x"]),
|
|
Instruction("test", ["eax", "eax"]),
|
|
Instruction("jnz", ["y"])
|
|
],
|
|
replace=[
|
|
Instruction("jg", ["y"])
|
|
]
|
|
),
|
|
PeepholeRule(
|
|
match=[
|
|
Instruction("setl",["al"]),
|
|
Instruction("movzx", ["rax", "al"]),
|
|
Instruction("push", ["rax"]),
|
|
Instruction("mov", ["eax", "x"]),
|
|
Instruction("test", ["eax", "eax"]),
|
|
Instruction("jnz", ["y"])
|
|
],
|
|
replace=[
|
|
Instruction("jl", ["y"])
|
|
]
|
|
),
|
|
PeepholeRule(
|
|
match=[
|
|
Instruction("sete",["al"]),
|
|
Instruction("movzx", ["rax", "al"]),
|
|
Instruction("push", ["rax"]),
|
|
Instruction("mov", ["eax", "x"]),
|
|
Instruction("test", ["eax", "eax"]),
|
|
Instruction("jnz", ["y"])
|
|
],
|
|
replace=[
|
|
Instruction("je", ["y"])
|
|
]
|
|
),
|
|
PeepholeRule(
|
|
match=[
|
|
Instruction("setne",["al"]),
|
|
Instruction("movzx", ["rax", "al"]),
|
|
Instruction("push", ["rax"]),
|
|
Instruction("mov", ["eax", "x"]),
|
|
Instruction("test", ["eax", "eax"]),
|
|
Instruction("jnz", ["y"])
|
|
],
|
|
replace=[
|
|
Instruction("jne", ["y"])
|
|
]
|
|
),
|
|
] |