datadevopscloud
diff --git a/‎compiler/src/compile.rs‎
Lines changed: 3 additions & 1 deletion b/‎compiler/src/compile.rs‎
Lines changed: 3 additions & 1 deletion
diff --git a/‎compiler/src/peephole.rs‎
Lines changed: 1 addition & 3 deletions b/‎compiler/src/peephole.rs‎
Lines changed: 1 addition & 3 deletions
diff --git a/‎compiler/src/peephole/optimizations.rs‎
Lines changed: 7 additions & 19 deletions b/‎compiler/src/peephole/optimizations.rs‎
Lines changed: 7 additions & 19 deletions
diff --git a/‎tests/snippets/dismod.py‎
Lines changed: 12 additions & 2 deletions b/‎tests/snippets/dismod.py‎
Lines changed: 12 additions & 2 deletions
@@ -848,7 +848,9 @@ impl<O: OutputStream> Compiler<O> {
 
         // Emit None at end:
         match body.last().map(|s| &s.node) {
-            Some(ast::StatementType::Return { .. }) => {}
+            Some(ast::StatementType::Return { .. }) => {
+                // the last instruction is a ReturnValue already, we don't need to emit it
+            }
             _ => {
                 self.emit(Instruction::LoadConst {
                     value: bytecode::Constant::None,
 
@@ -92,7 +92,7 @@ impl<O: OutputStream> PeepholeOptimizer<O> {
     }
 
     fn optimize(&mut self) {
-        apply_optimizations!(self, operator, unpack, useless_const);
+        apply_optimizations!(self, operator, unpack);
     }
 }
 
@@ -123,8 +123,6 @@ impl<O: OutputStream> OptimizationBuffer for PeepholeOptimizer<O> {
     }
 }
 
-// OPTIMIZATION
-
 pub trait OptimizationBuffer {
     fn emit(&mut self, instruction: Instruction, meta: InstructionMetadata);
     fn pop(&mut self) -> (Instruction, InstructionMetadata);
 
@@ -2,6 +2,11 @@ use rustpython_bytecode::bytecode::{self, Instruction};
 
 use super::{InstructionMetadata, OptimizationBuffer};
 
+macro_rules! metas {
+    [$($metas:expr),*$(,)?] => {
+        InstructionMetadata::from(vec![$($metas),*])
+    };
+}
 macro_rules! lc {
     ($name:ident {$($field:tt)*}) => {
         Instruction::LoadConst {
@@ -13,10 +18,10 @@ macro_rules! lc {
     };
 }
 macro_rules! emitconst {
-    ($buf:expr, [$($metas:expr),*], $($arg:tt)*) => {
+    ($buf:expr, [$($metas:expr),*$(,)?], $($arg:tt)*) => {
         $buf.emit(
             lc!($($arg)*),
-            InstructionMetadata::from(vec![$($metas),*]),
+            metas![$($metas),*],
         )
     };
 }
@@ -91,20 +96,3 @@ pub fn unpack(buf: &mut impl OptimizationBuffer) {
         buf.emit(instruction, meta)
     }
 }
-
-pub fn useless_const(buf: &mut impl OptimizationBuffer) {
-    let (instruction, meta) = buf.pop();
-    if instruction == Instruction::Pop {
-        let (arg, arg_meta) = buf.pop();
-        if let Instruction::LoadConst { .. } = arg {
-            // just ignore it all
-            drop(arg);
-            drop(instruction);
-        } else {
-            buf.emit(arg, arg_meta);
-            buf.emit(instruction, meta);
-        }
-    } else {
-        buf.emit(instruction, meta);
-    }
-}
@@ -2,9 +2,19 @@
 
 dis.dis(compile("5 + x + 5 or 2", "", "eval"))
 print("\n")
-dis.dis(compile("def f(x):\n   return 1", "", "exec"))
+dis.dis(compile("""
+def f(x):
+    return 1
+""", "", "exec"))
 print("\n")
-dis.dis(compile("if a:\n 1 or 2\nelif x == 'hello':\n 3\nelse:\n 4", "", "exec"))
+dis.dis(compile("""
+if a:
+    1 or 2
+elif x == 'hello':
+    3
+else:
+    4
+""", "", "exec"))
 print("\n")
 dis.dis(compile("f(x=1, y=2)", "", "eval"))
 print("\n")
Original file line number	Diff line number	Diff line change
`@@ -92,7 +92,7 @@ impl<O: OutputStream> PeepholeOptimizer<O> {`
`92`	`92`	`}`
`93`	`93`
`94`	`94`	`fn optimize(&mut self) {`
`95`		`- apply_optimizations!(self, operator, unpack, useless_const);`
	`95`	`+ apply_optimizations!(self, operator, unpack);`
`96`	`96`	`}`
`97`	`97`	`}`
`98`	`98`
`@@ -123,8 +123,6 @@ impl<O: OutputStream> OptimizationBuffer for PeepholeOptimizer<O> {`
`123`	`123`	`}`
`124`	`124`	`}`
`125`	`125`
`126`		`-// OPTIMIZATION`
`127`		`-`
`128`	`126`	`pub trait OptimizationBuffer {`
`129`	`127`	`fn emit(&mut self, instruction: Instruction, meta: InstructionMetadata);`
`130`	`128`	`fn pop(&mut self) -> (Instruction, InstructionMetadata);`