b76e849b5c
Extends avo to support most AVX-512 instruction sets.
The instruction type is extended to support suffixes. The K family of opmask
registers is added to the register package, and the operand package is updated
to support the new operand types. Move instruction deduction in `Load` and
`Store` is extended to support KMOV* and VMOV* forms.
Internal code generation packages were overhauled. Instruction database loading
required various messy changes to account for the additional complexities of the
AVX-512 instruction sets. The internal/api package was added to introduce a
separation between instruction forms in the database, and the functions avo
provides to create them. This was required since with instruction suffixes there
is no longer a one-to-one mapping between instruction constructors and opcodes.
AVX-512 bloated generated source code size substantially, initially increasing
compilation and CI test times to an unacceptable level. Two changes were made to
address this:
1. Instruction constructors in the `x86` package moved to an optab-based
approach. This compiles substantially faster than the verbose code
generation we had before.
2. The most verbose code-generated tests are moved under build tags and
limited to a stress test mode. Stress test builds are run on
schedule but not in regular CI.
An example of AVX-512 accelerated 16-lane MD5 is provided to demonstrate and
test the new functionality.
Updates #20 #163 #229
Co-authored-by: Vaughn Iverson <vsivsi@yahoo.com>
91 lines
1.9 KiB
Go
91 lines
1.9 KiB
Go
// Command avogen generates code based on the instruction database.
|
|
package main
|
|
|
|
import (
|
|
"flag"
|
|
"go/build"
|
|
"log"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"github.com/mmcloughlin/avo/internal/gen"
|
|
"github.com/mmcloughlin/avo/internal/inst"
|
|
"github.com/mmcloughlin/avo/internal/load"
|
|
"github.com/mmcloughlin/avo/printer"
|
|
)
|
|
|
|
var generators = map[string]gen.Builder{
|
|
"asmtest": gen.NewAsmTest,
|
|
"godata": gen.NewGoData,
|
|
"godatatest": gen.NewGoDataTest,
|
|
"optab": gen.NewOptab,
|
|
"ctors": gen.NewCtors,
|
|
"ctorstest": gen.NewCtorsTest,
|
|
"ctorsstress": gen.NewCtorsStress,
|
|
"ctorsbench": gen.NewCtorsBench,
|
|
"build": gen.NewBuild,
|
|
"buildtest": gen.NewBuildTest,
|
|
"mov": gen.NewMOV,
|
|
}
|
|
|
|
// Command-line flags.
|
|
var (
|
|
bootstrap = flag.Bool("bootstrap", false, "regenerate instruction list from original data")
|
|
datadir = flag.String(
|
|
"data",
|
|
filepath.Join(build.Default.GOPATH, "src/github.com/mmcloughlin/avo/internal/data"),
|
|
"path to data directory",
|
|
)
|
|
output = flag.String("output", "", "path to output file (default stdout)")
|
|
)
|
|
|
|
func main() {
|
|
log.SetPrefix("avogen: ")
|
|
log.SetFlags(0)
|
|
flag.Parse()
|
|
|
|
// Build generator.
|
|
t := flag.Arg(0)
|
|
builder := generators[t]
|
|
if builder == nil {
|
|
log.Fatalf("unknown generator type '%s'", t)
|
|
}
|
|
|
|
g := builder(printer.NewArgvConfig())
|
|
|
|
// Determine output writer.
|
|
w := os.Stdout
|
|
if *output != "" {
|
|
f, err := os.Create(*output)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
defer f.Close()
|
|
w = f
|
|
}
|
|
|
|
// Load instructions.
|
|
is := inst.Instructions
|
|
if *bootstrap {
|
|
log.Printf("bootstrap: loading instructions from data directory %s", *datadir)
|
|
l := load.NewLoaderFromDataDir(*datadir)
|
|
r, err := l.Load()
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
is = r
|
|
}
|
|
|
|
// Generate output.
|
|
b, generr := g.Generate(is)
|
|
|
|
// Write.
|
|
if _, err := w.Write(b); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
if generr != nil {
|
|
log.Fatal(generr)
|
|
}
|
|
}
|