1 files changed, 95 insertions, 0 deletions
diff --git a/src/cmd/compile/internal/ssa/sccp_test.go b/src/cmd/compile/internal/ssa/sccp_test.go
new file mode 100644
index 0000000..70c23e7
--- /dev/null
+++ b/src/cmd/compile/internal/ssa/sccp_test.go
@@ -0,0 +1,95 @@
+// Copyright 2023 The Go Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style
+// license that can be found in the LICENSE file.
+
+package ssa
+
+import (
+	"cmd/compile/internal/types"
+	"strings"
+	"testing"
+)
+
+func TestSCCPBasic(t *testing.T) {
+	c := testConfig(t)
+	fun := c.Fun("b1",
+		Bloc("b1",
+			Valu("mem", OpInitMem, types.TypeMem, 0, nil),
+			Valu("v1", OpConst64, c.config.Types.Int64, 20, nil),
+			Valu("v2", OpConst64, c.config.Types.Int64, 21, nil),
+			Valu("v3", OpConst64F, c.config.Types.Float64, 21.0, nil),
+			Valu("v4", OpConstBool, c.config.Types.Bool, 1, nil),
+			Valu("t1", OpAdd64, c.config.Types.Int64, 0, nil, "v1", "v2"),
+			Valu("t2", OpDiv64, c.config.Types.Int64, 0, nil, "t1", "v1"),
+			Valu("t3", OpAdd64, c.config.Types.Int64, 0, nil, "t1", "t2"),
+			Valu("t4", OpSub64, c.config.Types.Int64, 0, nil, "t3", "v2"),
+			Valu("t5", OpMul64, c.config.Types.Int64, 0, nil, "t4", "v2"),
+			Valu("t6", OpMod64, c.config.Types.Int64, 0, nil, "t5", "v2"),
+			Valu("t7", OpAnd64, c.config.Types.Int64, 0, nil, "t6", "v2"),
+			Valu("t8", OpOr64, c.config.Types.Int64, 0, nil, "t7", "v2"),
+			Valu("t9", OpXor64, c.config.Types.Int64, 0, nil, "t8", "v2"),
+			Valu("t10", OpNeg64, c.config.Types.Int64, 0, nil, "t9"),
+			Valu("t11", OpCom64, c.config.Types.Int64, 0, nil, "t10"),
+			Valu("t12", OpNeg64, c.config.Types.Int64, 0, nil, "t11"),
+			Valu("t13", OpFloor, c.config.Types.Float64, 0, nil, "v3"),
+			Valu("t14", OpSqrt, c.config.Types.Float64, 0, nil, "t13"),
+			Valu("t15", OpCeil, c.config.Types.Float64, 0, nil, "t14"),
+			Valu("t16", OpTrunc, c.config.Types.Float64, 0, nil, "t15"),
+			Valu("t17", OpRoundToEven, c.config.Types.Float64, 0, nil, "t16"),
+			Valu("t18", OpTrunc64to32, c.config.Types.Int64, 0, nil, "t12"),
+			Valu("t19", OpCvt64Fto64, c.config.Types.Float64, 0, nil, "t17"),
+			Valu("t20", OpCtz64, c.config.Types.Int64, 0, nil, "v2"),
+			Valu("t21", OpSlicemask, c.config.Types.Int64, 0, nil, "t20"),
+			Valu("t22", OpIsNonNil, c.config.Types.Int64, 0, nil, "v2"),
+			Valu("t23", OpNot, c.config.Types.Bool, 0, nil, "v4"),
+			Valu("t24", OpEq64, c.config.Types.Bool, 0, nil, "v1", "v2"),
+			Valu("t25", OpLess64, c.config.Types.Bool, 0, nil, "v1", "v2"),
+			Valu("t26", OpLeq64, c.config.Types.Bool, 0, nil, "v1", "v2"),
+			Valu("t27", OpEqB, c.config.Types.Bool, 0, nil, "v4", "v4"),
+			Valu("t28", OpLsh64x64, c.config.Types.Int64, 0, nil, "v2", "v1"),
+			Valu("t29", OpIsInBounds, c.config.Types.Int64, 0, nil, "v2", "v1"),
+			Valu("t30", OpIsSliceInBounds, c.config.Types.Int64, 0, nil, "v2", "v1"),
+			Goto("b2")),
+		Bloc("b2",
+			Exit("mem")))
+	sccp(fun.f)
+	CheckFunc(fun.f)
+	for name, value := range fun.values {
+		if strings.HasPrefix(name, "t") {
+			if !isConst(value) {
+				t.Errorf("Must be constant: %v", value.LongString())
+			}
+		}
+	}
+}
+
+func TestSCCPIf(t *testing.T) {
+	c := testConfig(t)
+	fun := c.Fun("b1",
+		Bloc("b1",
+			Valu("mem", OpInitMem, types.TypeMem, 0, nil),
+			Valu("v1", OpConst64, c.config.Types.Int64, 0, nil),
+			Valu("v2", OpConst64, c.config.Types.Int64, 1, nil),
+			Valu("cmp", OpLess64, c.config.Types.Bool, 0, nil, "v1", "v2"),
+			If("cmp", "b2", "b3")),
+		Bloc("b2",
+			Valu("v3", OpConst64, c.config.Types.Int64, 3, nil),
+			Goto("b4")),
+		Bloc("b3",
+			Valu("v4", OpConst64, c.config.Types.Int64, 4, nil),
+			Goto("b4")),
+		Bloc("b4",
+			Valu("merge", OpPhi, c.config.Types.Int64, 0, nil, "v3", "v4"),
+			Exit("mem")))
+	sccp(fun.f)
+	CheckFunc(fun.f)
+	for _, b := range fun.blocks {
+		for _, v := range b.Values {
+			if v == fun.values["merge"] {
+				if !isConst(v) {
+					t.Errorf("Must be constant: %v", v.LongString())
+				}
+			}
+		}
+	}
+}