mirrors/kitty
1
0
Fork 0
mirror of https://github.com/kovidgoyal/kitty synced 2026-07-09 05:35:19 +02:00
Code Issues Packages Projects Releases Wiki Activity

Add an IndexByte implementation useful for benchmarking against stdlib SIMD implementation

Browse Source
This commit is contained in:
Kovid Goyal
2024-03-07 09:36:40 +05:30
parent 9ea0dde469
commit 47fea26b62
4 changed files with 71 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
tools/simdstring/generate.go
View File

@@ -1451,6 +1451,27 @@ func (s *State) indexbyte2() {
}
func (s *State) indexbyte_body(f *Function) {
b := f.Vec()
f.Set1Epi8("b", b)
test_bytes := func(bytes_to_test, test_ans Register) {
f.CmpEqEpi8(bytes_to_test, b, test_ans)
}
s.index_func(f, test_bytes)
}
func (s *State) indexbyte() {
f := s.NewFunction("index_byte_asm", "Find the index of a byte", []FunctionParam{{"data", ByteSlice}, {"b", types.Byte}}, []FunctionParam{{"ans", types.Int}})
if s.ISA.HasSIMD {
s.indexbyte_body(f)
}
f = s.NewFunction("index_byte_string_asm", "Find the index of a byte", []FunctionParam{{"data", types.String}, {"b", types.Byte}}, []FunctionParam{{"ans", types.Int}})
if s.ISA.HasSIMD {
s.indexbyte_body(f)
}
}
func (s *State) indexc0_body(f *Function) {
lower := f.Vec()
upper := f.Vec()
@@ -1494,6 +1515,7 @@ func (s *State) Generate() {
s.indexbyte2()
s.indexc0()
s.indexbyte()
s.OutputFunction()
}