我试图写一些SIMD主要用于学习目的。我知道Go可以链接程序集,但是我无法使其正常工作。
这是我可以做的最简单的示例(逐元素矢量乘法):
vec_amd64.s(注意:实际文件下面有一个空格行,RET
因为否则会导致错误)
// func mul(v1, v2 Vec4) Vec4
TEXT .mul(SB),4,$0-48
MOVUPS v1+0(FP), X0
MOVUPS v2+16(FP), X1
MULPS X1, X0
// also tried ret+32 since I've seen some places do that
MOVUPS X0, toReturn+32(FP)
RET
vec.go
package simd
type Vec4 [4]float32
func (v1 Vec4) Mul(v2 Vec4) Vec4 {
return Vec4{v1[0] * v2[0], v1[1] * v2[1], v1[2] * v2[2], v1[3] * v2[3]}
}
func mul(v1, v2 Vec4) Vec4
simd_test.go
package simd
import (
"testing"
)
func TestMul(t *testing.T) {
v1 := Vec4{1, 2, 3, 4}
v2 := Vec4{5, 6, 7, 8}
res := v1.Mul(v2)
res2 := mul(v1, v2)
// Placeholder until I get it to compile
if res != res2 {
t.Fatalf("Expected %v; got %v", res, res2)
}
}
当我尝试运行时go test
,出现错误:
# testmain
simd.TestMul: call to external function simd.mul
simd.TestMul: undefined: simd.mul
该go env
命令报告myGOHOSTARCH
为,amd64
而Go版本为1.3。为了确认这不是导致此问题的体系结构,我找到了另一个使用程序集的程序包,并删除了所有程序集文件,除了_amd64.s
一个程序包及其测试运行良好。
我还尝试将其更改为导出的标识符,以防引起怪异但没有骰子。我想我非常密切地关注了像math/big
这样的软件包中的模板,因此希望它是我所缺少的简单明了的东西。
我知道Go至少会尝试使用该程序集,因为如果我在.s文件中引入语法错误,则构建工具会抱怨它。
编辑:
要清楚,go build
将干净地编译,但go test
会导致错误出现。
go build
干净地完成,go test
失败。