91 lines
1.7 KiB
Go
91 lines
1.7 KiB
Go
package pass_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/mmcloughlin/avo/reg"
|
|
|
|
"github.com/mmcloughlin/avo/pass"
|
|
|
|
"github.com/mmcloughlin/avo"
|
|
"github.com/mmcloughlin/avo/build"
|
|
"github.com/mmcloughlin/avo/operand"
|
|
)
|
|
|
|
func TestLivenessBasic(t *testing.T) {
|
|
// Build: a = 1, b = 2, a = a+b
|
|
ctx := build.NewContext()
|
|
ctx.Function("add")
|
|
a := ctx.GP64v()
|
|
b := ctx.GP64v()
|
|
ctx.MOVQ(operand.Imm(1), a)
|
|
ctx.MOVQ(operand.Imm(2), b)
|
|
ctx.ADDQ(a, b)
|
|
|
|
AssertLiveness(t, ctx,
|
|
[][]reg.Register{
|
|
{},
|
|
{a},
|
|
{a, b},
|
|
},
|
|
[][]reg.Register{
|
|
{a},
|
|
{a, b},
|
|
{},
|
|
},
|
|
)
|
|
}
|
|
|
|
func AssertLiveness(t *testing.T, ctx *build.Context, in, out [][]reg.Register) {
|
|
fn := ConstructLiveness(t, ctx)
|
|
is := fn.Instructions()
|
|
|
|
if len(in) != len(is) || len(out) != len(is) {
|
|
t.Fatalf("%d instructions: %d/%d in/out expectations", len(is), len(in), len(out))
|
|
}
|
|
|
|
for idx, i := range is {
|
|
AssertRegistersMatchSet(t, in[idx], i.LiveIn)
|
|
AssertRegistersMatchSet(t, out[idx], i.LiveOut)
|
|
}
|
|
}
|
|
|
|
func AssertRegistersMatchSet(t *testing.T, rs []reg.Register, s map[reg.ID]bool) {
|
|
if len(rs) != len(s) {
|
|
t.Fatalf("size mismatch")
|
|
}
|
|
for _, r := range rs {
|
|
if _, found := s[r.ID()]; !found {
|
|
t.Fatalf("missing register ID %v", r.ID())
|
|
}
|
|
}
|
|
}
|
|
|
|
func ConstructLiveness(t *testing.T, ctx *build.Context) *avo.Function {
|
|
f, errs := ctx.Result()
|
|
if errs != nil {
|
|
for _, err := range errs {
|
|
t.Error(err)
|
|
}
|
|
t.FailNow()
|
|
}
|
|
|
|
if len(f.Functions) != 1 {
|
|
t.Fatalf("expect 1 function")
|
|
}
|
|
fn := f.Functions[0]
|
|
|
|
passes := []func(*avo.Function) error{
|
|
pass.LabelTarget,
|
|
pass.CFG,
|
|
pass.Liveness,
|
|
}
|
|
for _, p := range passes {
|
|
if err := p(fn); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
return fn
|
|
}
|