Gensimd is a collection of packages and a command, gensimd, for using SIMD in Go.
Write functions in a subset of Go, using the gensimd/simd, gensimd/simd/sse2
packages and convert them to Go assembly with the gensimd command.
CODEGEN REWRITE - Code generation is being rewritten to use the new Go compiler SSA backend, the rewrite is in progress at https://github.com/bjwbell/ssa and https://github.com/bjwbell/gir.
export GO15VENDOREXPERIMENT=1
go get github.com/bjwbell/gensimd
go install github.com/bjwbell/gensimd
go get github.com/bjwbell/gensimd/simd
The SSE2 intrinsics package is github.com/bjwbell/gensimd/simd/sse
To build and run all examples execute ./run_examples.sh.
To build and run the reference tests execute ./run_tests.sh.
[bjwbell]$ gensimd --help
-debug
include debug comments in assembly
-f string
input file with function definitions
-fn string
comma separated list of function names
-goprotofile string
output file for SIMD function prototype(s)
-o string
Go assembly output file
-outfn string
comma separated list of output function names
-spills
print each register spill
-ssa
dump ssa representation
For functions gensimd translates from Go to assembly it supports only a small subset of Go.
- Integers and floats -
uint8/int8,uint16/int16,uint32/int32,uint64/int64,float32/float64 ifstatements,forloops (except withrange)- Arrays and slices
- Heap allocated local variables
- Multiple and named return values
- Builtins except
len - Function calls except to
simd.* - Method calls
- All struct types except
simd.* - Keywords
range,map,select,chan,defer - Slice creation e.g.
newslice := slice[1:len(slice) - 2]
- Slice access bounds checking
SIMD intrinsics are availabe if simd.Available() returns true.
The common vs platform specific functionality is inspired from Huon Wilson's SIMD work.
For unsigned integer values, the SIMD functions Add*, Sub*, Mul*, and Shl* are computed modulo 2^n, where n is the bit width of the unsigned integer's type. These unsigned integer operations discard high bits upon overflow.
For signed integers, the SIMD functions Add*, Sub*, Mul*, and Shl* may overflow and the resulting value is defined by the signed integer representation, the operation, and its operands. The behavior is guaranteed to be identical to the Go versions in gensimd/simd/simd.go.
For both unsigned and signed integer values, the SIMD function Shr* is guaranteed to have the same behavior as the Go version in gensimd/simd/simd.go
The behavior of the floating point SIMD functions Add*, Sub*, Mul*, and Div* is guaranteed to be identical to the Go versions in gensimd/simd/simd.go.
type I8x16 [16]int8
type I16x8 [8]int16
type I32x4 [4]int32
type I64x2 [2]int64
type U8x16 [16]uint8
type U16x8 [8]uint16
type U32x4 [4]uint32
type U64x2 [2]uint64
type F32x4 [4]float32
type F64x2 [2]float64
func AddI8x16(x, y I8x16) I8x16
func SubI8x16(x, y I8x16) I8x16
func AddU8x16(x, y U8x16) U8x16
func SubU8x16(x, y U8x16) U8x16
func AddI16x8(x, y I16x8) I16x8
func SubI16x8(x, y I16x8) I16x8
func MulI16x8(x, y I16x8) I16x8
func ShlI16x8(x I16x8, shift uint8) I16x8
func ShrI16x8(x I16x8, shift uint8) I16x8
func AddU16x8(x, y U16x8) U16x8
func SubU16x8(x, y U16x8) U16x8
func MulU16x8(x, y U16x8) U16x8
func ShlU16x8(x U16x8, shift uint8) U16x8
func ShrU16x8(x U16x8, shift uint8) U16x8
func AddI32x4(x, y I32x4) I32x4
func SubI32x4(x, y I32x4) I32x4
func MulI32x4(x, y I32x4) I32x4
func ShlI32x4(x I32x4, shift uint8) I32x4
func ShrI32x4(x I32x4, shift uint8) I32x4
func ShuffleI32x4(x I32x4, order uint8) I32x4
func AddU32x4(x, y U32x4) U32x4
func SubU32x4(x, y U32x4) U32x4
func MulU32x4(x, y U32x4) U32x4
func ShlU32x4(x U32x4, shift uint8) U32x4
func ShrU32x4(x U32x4, shift uint8) U32x4
func ShuffleU32x4(x U32x4, order uint8) U32x4
func AddI64x2(x, y I64x2) I64x2
func SubI64x2(x, y I64x2) I64x2
func AddU64x2(x, y U64x2) U64x2
func SubU64x2(x, y U64x2) U64x2
func AddF32x4(x, y F32x4) F32x4
func SubF32x4(x, y F32x4) F32x4
func MulF32x4(x, y F32x4) F32x4
func DivF32x4(x, y F32x4) F32x4
func AddF64x2(x, y F64x2) F64x2
func SubF64x2(x, y F64x2) F64x2
func MulF64x2(x, y F64x2) F64x2
func DivF64x2(x, y F64x2) F64x2
There are no SIMD functions for 64 bit integer multiplication because there's no equivalent SSE2 instruction.
Until Go 1.6, ShrU16x8 is slow because of golang/go#13010 "cmd/asm: x86, incorrect Optab entry - PSRLW".
MulI32x4 is slow because the instruction "PMULLD" wasn't added until SSE4.1.
It's emulated using SSE2 instructions, SSE multiplication of 4 32-bit integers.
The shuffle order operand in ShuffleI32x4/ShuffleU32x4 MUST be a constant not a variable. Example:
const order uint8 = 8
simd.SuffleU32x4(x, order)
The below functions aren't implemented because they have no directly equivalent SSE2 instructions.
func ShlI64x2(x, shift uint8) I64x2
func ShrI64x2(x, shift uint8) I64x2
func ShlU64x2(x, shift uint8) U64x2
func ShrU64x2(x, shift uint8) U64x2
SSE2 intrinsics are availabe if simd.SSE2() returns true.
type M128i [16]byte
type M128 [4]float32
type M128d [2]float64
func AddEpi64(x, y simd.M128i) simd.M128i
func SubEpi64(x, y simd.M128i) simd.M128i
func MulEpu32(x, y simd.M128i) simd.M128i
func ShufflehiEpi16(x, y simd.M128i) simd.M128i
func ShuffleloEpi16(x, y simd.M128i) simd.M128i
func ShuffleEpi32(x, y simd.M128i) simd.M128i
func SlliSi128(x, y simd.M128i) simd.M128i
func SrliSi128(x, y simd.M128i) simd.M128i
func UnpackhiEpi64(x, y simd.M128i) simd.M128i
func UnpackloEpi64(x, y simd.M128i) simd.M128i
func AddPd(x, y simd.M128d) simd.M128d
func AddSd(x, y simd.M128d) simd.M128d
func AndnotPd(x, y simd.M128d) simd.M128d
func CmpeqPd(x, y simd.M128d) simd.M128d
func CmpeqSd(x, y simd.M128d) simd.M128d