mirror of
https://github.com/kovidgoyal/kitty.git
synced 2026-02-01 11:34:59 +01:00
65 lines
2.0 KiB
Go
65 lines
2.0 KiB
Go
// License: GPLv3 Copyright: 2024, Kovid Goyal, <kovid at kovidgoyal.net>
|
|
|
|
package simdstring
|
|
|
|
import (
|
|
"runtime"
|
|
|
|
"golang.org/x/sys/cpu"
|
|
)
|
|
|
|
var Have128bit = false
|
|
var Have256bit = false
|
|
var VectorSize = 1
|
|
|
|
// Return the index at which b first occurs in data. If not found -1 is returned.
|
|
var IndexByte func(data []byte, b byte) int = index_byte_scalar
|
|
|
|
// Return the index at which either a or b first occurs in text. If neither is
|
|
// found -1 is returned.
|
|
var IndexByteString func(text string, b byte) int = index_byte_string_scalar
|
|
|
|
// Return the index at which either a or b first occurs in data. If neither is
|
|
// found -1 is returned.
|
|
var IndexByte2 func(data []byte, a, b byte) int = index_byte2_scalar
|
|
|
|
// Return the index at which either a or b first occurs in text. If neither is
|
|
// found -1 is returned.
|
|
var IndexByte2String func(text string, a, b byte) int = index_byte2_string_scalar
|
|
|
|
// Return the index at which the first C0 byte is found or -1 when no such bytes are present.
|
|
var IndexC0 func(data []byte) int = index_c0_scalar
|
|
|
|
// Return the index at which the first C0 byte is found or -1 when no such bytes are present.
|
|
var IndexC0String func(data string) int = index_c0_string_scalar
|
|
|
|
func init() {
|
|
switch runtime.GOARCH {
|
|
case "amd64":
|
|
if cpu.Initialized {
|
|
Have128bit = cpu.X86.HasSSE42 && HasSIMD128Code
|
|
Have256bit = cpu.X86.HasAVX2 && HasSIMD256Code
|
|
}
|
|
case "arm64":
|
|
Have128bit = HasSIMD128Code
|
|
Have256bit = HasSIMD256Code
|
|
}
|
|
if Have256bit {
|
|
IndexByte = index_byte_asm_256
|
|
IndexByteString = index_byte_string_asm_256
|
|
IndexByte2 = index_byte2_asm_256
|
|
IndexByte2String = index_byte2_string_asm_256
|
|
IndexC0 = index_c0_asm_256
|
|
IndexC0String = index_c0_string_asm_256
|
|
VectorSize = 32
|
|
} else if Have128bit {
|
|
IndexByte = index_byte_asm_128
|
|
IndexByteString = index_byte_string_asm_128
|
|
IndexByte2 = index_byte2_asm_128
|
|
IndexByte2String = index_byte2_string_asm_128
|
|
IndexC0 = index_c0_asm_128
|
|
IndexC0String = index_c0_string_asm_128
|
|
VectorSize = 16
|
|
}
|
|
}
|