Skip to content

Split stack_effect into pushed & popped #6893

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
youknowone merged 13 commits into from
Jan 31, 2026
Merged

Split stack_effect into pushed & popped #6893

Show file tree
Hide file tree
Changes from 2 commits
Commits
Show all changes
13 commits
Select commit Hold shift + click to select a range
52b65a8
Split stack_effect into pushed & popped
ShaharNaveh Jan 27, 2026
654609b
Fix typo
ShaharNaveh Jan 29, 2026
e407978
Merge remote-tracking branch 'upstream/main' into opcodes-stack-effect
ShaharNaveh Jan 30, 2026
1ea2bd9
Fix merge
ShaharNaveh Jan 30, 2026
97662d0
Make `StructEffect` a struct
ShaharNaveh Jan 30, 2026
5951499
Fix more code
ShaharNaveh Jan 30, 2026
b355782
Merge remote-tracking branch 'upstream/main' into opcodes-stack-effect
ShaharNaveh Jan 30, 2026
bf92d5c
Fix typo
ShaharNaveh Jan 30, 2026
1f92b11
Merge remote-tracking branch 'upstream/main' into opcodes-stack-effect
ShaharNaveh Jan 30, 2026
9f4d512
Simpify `stack_effect_info`
ShaharNaveh Jan 30, 2026
b5587ac
Fix code
ShaharNaveh Jan 30, 2026
b4e8dea
Simplfy casting as checks
ShaharNaveh Jan 30, 2026
2137406
Merge remote-tracking branch 'upstream/main' into opcodes-stack-effect
ShaharNaveh Jan 30, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 3 additions & 5 deletions crates/codegen/src/ir.rs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@ use rustpython_compiler_core::{
bytecode::{
AnyInstruction, Arg, CodeFlags, CodeObject, CodeUnit, CodeUnits, ConstantData,
ExceptionTableEntry, InstrDisplayContext, Instruction, InstructionMetadata, Label, OpArg,
PseudoInstruction, PyCodeLocationInfoKind, encode_exception_table,
PseudoInstruction, PyCodeLocationInfoKind, StackEffect, encode_exception_table,
},
varint::{write_signed_varint, write_varint},
};
Expand Down Expand Up @@ -697,8 +697,6 @@ impl CodeInfo {
/// This is a reference counting optimization in CPython; in RustPython
/// we implement it for bytecode compatibility.
fn optimize_load_fast_borrow(&mut self) {
use rustpython_compiler_core::bytecode::InstructionMetadata;

// NOT_LOCAL marker: instruction didn't come from a LOAD_FAST
const NOT_LOCAL: usize = usize::MAX;

Expand Down Expand Up @@ -733,7 +731,7 @@ impl CodeInfo {
// is pop 2 push 1, not pop 1 push 0). We list those explicitly;
// the fallback under-pops and under-pushes, which is conservative
// (may miss optimisation opportunities but never miscompiles).
let effect = instr.stack_effect(info.arg);
let effect = instr.stack_effect(u32::from(info.arg) as i32);
Comment thread
coderabbitai[bot] marked this conversation as resolved.
Outdated
let (pops, pushes) = match instr {
// --- pop 2, push 1 ---
Instruction::BinaryOp { .. }
Expand Down Expand Up @@ -866,7 +864,7 @@ impl CodeInfo {
let block = &self.blocks[block_idx];
for ins in &block.instructions {
let instr = &ins.instr;
let effect = instr.stack_effect(ins.arg);
let effect = instr.stack_effect(u32::from(ins.arg) as i32);
if DEBUG {
let display_arg = if ins.target == BlockIdx::NULL {
ins.arg
Expand Down
2 changes: 1 addition & 1 deletion crates/compiler-core/src/bytecode.rs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@ use rustpython_wtf8::{Wtf8, Wtf8Buf};

pub use crate::bytecode::{
instruction::{
AnyInstruction, Arg, Instruction, InstructionMetadata, PseudoInstruction,
AnyInstruction, Arg, Instruction, InstructionMetadata, PseudoInstruction, StackEffect,
decode_load_attr_arg, decode_load_super_attr_arg, encode_load_attr_arg,
encode_load_super_attr_arg,
},
Expand Down
Loading
Loading