diff --git a/src/ir_opt/constant_propagation_pass.cpp b/src/ir_opt/constant_propagation_pass.cpp index 10f1d720..a8018711 100644 --- a/src/ir_opt/constant_propagation_pass.cpp +++ b/src/ir_opt/constant_propagation_pass.cpp @@ -123,6 +123,15 @@ void FoldSignExtendXToWord(IR::Inst& inst) { inst.ReplaceUsesWith(IR::Value{static_cast(value)}); } +void FoldSignExtendXToLong(IR::Inst& inst) { + if (!inst.AreAllArgsImmediates()) { + return; + } + + const s64 value = inst.GetArg(0).GetImmediateAsS64(); + inst.ReplaceUsesWith(IR::Value{static_cast(value)}); +} + void FoldZeroExtendXToWord(IR::Inst& inst) { if (!inst.AreAllArgsImmediates()) { return; @@ -185,6 +194,11 @@ void ConstantPropagation(IR::Block& block) { case IR::Opcode::SignExtendHalfToWord: FoldSignExtendXToWord(inst); break; + case IR::Opcode::SignExtendByteToLong: + case IR::Opcode::SignExtendHalfToLong: + case IR::Opcode::SignExtendWordToLong: + FoldSignExtendXToLong(inst); + break; case IR::Opcode::ZeroExtendByteToWord: case IR::Opcode::ZeroExtendHalfToWord: FoldZeroExtendXToWord(inst);