Add support for stack-only constraints in fastalloc

d-sonuga · d-sonuga · commit 239181b903f6 · 2025-08-24T18:46:09.000+01:00
diff --git a/src/fastalloc/mod.rs b/src/fastalloc/mod.rs
@@ -508,9 +508,7 @@ impl<'a, F: Function> Env<'a, F> {
                 unreachable!()
             }
 
-            OperandConstraint::Stack => {
-                panic!("Stack constraints not supported in fastalloc");
-            }
+            OperandConstraint::Stack => self.edits.is_stack(alloc),
         }
     }
 
@@ -668,9 +666,7 @@ impl<'a, F: Function> Env<'a, F> {
                 unreachable!();
             }
 
-            OperandConstraint::Stack => {
-                panic!("Stack operand constraints not supported in fastalloc");
-            }
+            OperandConstraint::Stack => Allocation::stack(self.get_spillslot(op.vreg())),
         };
         self.allocs[(inst.index(), op_idx)] = new_alloc;
         Ok(new_alloc)

Original file line number	Diff line number	Diff line change
`@@ -508,9 +508,7 @@ impl<'a, F: Function> Env<'a, F> {`
`508`	`508`	`unreachable!()`
`509`	`509`	`}`
`510`	`510`
`511`		`- OperandConstraint::Stack => {`
`512`		`- panic!("Stack constraints not supported in fastalloc");`
`513`		`- }`
	`511`	`+ OperandConstraint::Stack => self.edits.is_stack(alloc),`
`514`	`512`	`}`
`515`	`513`	`}`
`516`	`514`
`@@ -668,9 +666,7 @@ impl<'a, F: Function> Env<'a, F> {`
`668`	`666`	`unreachable!();`
`669`	`667`	`}`
`670`	`668`
`671`		`- OperandConstraint::Stack => {`
`672`		`- panic!("Stack operand constraints not supported in fastalloc");`
`673`		`- }`
	`669`	`+ OperandConstraint::Stack => Allocation::stack(self.get_spillslot(op.vreg())),`
`674`	`670`	`};`
`675`	`671`	`self.allocs[(inst.index(), op_idx)] = new_alloc;`
`676`	`672`	`Ok(new_alloc)`