Tobias Klauser
d6449816ce
CL 106235 removed runtime.support_avx on which blake2b depends to determine AVX support. Switch to use the corresponding feature flags from internal/cpu instead. While at it, also use AVX2 and SSE4 (for Go 1.7 and later) from there. Change-Id: I4f7a27fcfa0c5d0d7a50444c7aa32519b97e90dd Reviewed-on: https://go-review.googlesource.com/106336 Run-TryBot: Tobias Klauser <tobias.klauser@gmail.com> TryBot-Result: Gobot Gobot <gobot@golang.org> Reviewed-by: Brad Fitzpatrick <bradfitz@golang.org>
46 lines
1.1 KiB
Go
46 lines
1.1 KiB
Go
// Copyright 2016 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.
|
|
|
|
// +build go1.7,amd64,!gccgo,!appengine
|
|
|
|
package blake2b
|
|
|
|
import _ "unsafe"
|
|
|
|
//go:linkname x86_HasAVX internal/cpu.X86.HasAVX
|
|
var x86_HasAVX bool
|
|
|
|
//go:linkname x86_HasAVX2 internal/cpu.X86.HasAVX2
|
|
var x86_HasAVX2 bool
|
|
|
|
//go:linkname x86_HasAVX internal/cpu.X86.HasSSE4
|
|
var x86_HasSSE4 bool
|
|
|
|
func init() {
|
|
useAVX2 = x86_HasAVX2
|
|
useAVX = x86_HasAVX
|
|
useSSE4 = x86_HasSSE4
|
|
}
|
|
|
|
//go:noescape
|
|
func hashBlocksAVX2(h *[8]uint64, c *[2]uint64, flag uint64, blocks []byte)
|
|
|
|
//go:noescape
|
|
func hashBlocksAVX(h *[8]uint64, c *[2]uint64, flag uint64, blocks []byte)
|
|
|
|
//go:noescape
|
|
func hashBlocksSSE4(h *[8]uint64, c *[2]uint64, flag uint64, blocks []byte)
|
|
|
|
func hashBlocks(h *[8]uint64, c *[2]uint64, flag uint64, blocks []byte) {
|
|
if useAVX2 {
|
|
hashBlocksAVX2(h, c, flag, blocks)
|
|
} else if useAVX {
|
|
hashBlocksAVX(h, c, flag, blocks)
|
|
} else if useSSE4 {
|
|
hashBlocksSSE4(h, c, flag, blocks)
|
|
} else {
|
|
hashBlocksGeneric(h, c, flag, blocks)
|
|
}
|
|
}
|