3 // Copyright 2017 The Go Authors. All rights reserved.
4 // Use of this source code is governed by a BSD-style
5 // license that can be found in the LICENSE file.
11 // Check that expressions like (c*n + d*(n+k)) get correctly merged by
12 // the compiler into (c+d)*n + d*k (with c+d and d*k computed at
15 // The merging is performed by a combination of the multiplication
17 // (c*n + d*n) -> (c+d)*n
18 // and the distributive multiplication rules
19 // c * (d+x) -> c*d + c*x
21 // Generate a MergeTest that looks like this:
23 // a8, b8 = m1*n8 + m2*(n8+k), (m1+m2)*n8 + m2*k
25 // // print error msg and panic
27 func makeMergeAddTest(m1, m2, k int, size string) string {
29 model := " a" + size + ", b" + size
30 model += fmt.Sprintf(" = %%d*n%s + %%d*(n%s+%%d), (%%d+%%d)*n%s + (%%d*%%d)", size, size, size)
32 test := fmt.Sprintf(model, m1, m2, k, m1, m2, m2, k)
35 fmt.Printf("MergeAddTest(%d, %d, %d, %s) failed\n")
36 fmt.Printf("%%d != %%d\n", a%s, b%s)
39 `, size, size, m1, m2, k, size, size, size)
43 // Check that expressions like (c*n - d*(n+k)) get correctly merged by
44 // the compiler into (c-d)*n - d*k (with c-d and d*k computed at
47 // The merging is performed by a combination of the multiplication
49 // (c*n - d*n) -> (c-d)*n
50 // and the distributive multiplication rules
51 // c * (d-x) -> c*d - c*x
53 // Generate a MergeTest that looks like this:
55 // a8, b8 = m1*n8 - m2*(n8+k), (m1-m2)*n8 - m2*k
57 // // print error msg and panic
59 func makeMergeSubTest(m1, m2, k int, size string) string {
61 model := " a" + size + ", b" + size
62 model += fmt.Sprintf(" = %%d*n%s - %%d*(n%s+%%d), (%%d-%%d)*n%s - (%%d*%%d)", size, size, size)
64 test := fmt.Sprintf(model, m1, m2, k, m1, m2, m2, k)
67 fmt.Printf("MergeSubTest(%d, %d, %d, %s) failed\n")
68 fmt.Printf("%%d != %%d\n", a%s, b%s)
71 `, size, size, m1, m2, k, size, size, size)
75 func makeAllSizes(m1, m2, k int) string {
77 tests += makeMergeAddTest(m1, m2, k, "8")
78 tests += makeMergeAddTest(m1, m2, k, "16")
79 tests += makeMergeAddTest(m1, m2, k, "32")
80 tests += makeMergeAddTest(m1, m2, k, "64")
81 tests += makeMergeSubTest(m1, m2, k, "8")
82 tests += makeMergeSubTest(m1, m2, k, "16")
83 tests += makeMergeSubTest(m1, m2, k, "32")
84 tests += makeMergeSubTest(m1, m2, k, "64")
90 fmt.Println(`package main
106 fmt.Println(makeAllSizes(03, 05, 0)) // 3*n + 5*n
107 fmt.Println(makeAllSizes(17, 33, 0))
108 fmt.Println(makeAllSizes(80, 45, 0))
109 fmt.Println(makeAllSizes(32, 64, 0))
111 fmt.Println(makeAllSizes(7, 11, +1)) // 7*n + 11*(n+1)
112 fmt.Println(makeAllSizes(9, 13, +2))
113 fmt.Println(makeAllSizes(11, 16, -1))
114 fmt.Println(makeAllSizes(17, 9, -2))