mirrors/kitty
1
0
Fork 0
mirror of https://github.com/kovidgoyal/kitty synced 2026-07-20 23:44:59 +02:00
Code Issues Packages Projects Releases Wiki Activity

DRYer

Browse Source
This commit is contained in:
Kovid Goyal
2024-02-11 06:59:42 +05:30
parent b5edd9ad57
commit bbdb0b15f3
1 changed files with 22 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

39
tools/simdstring/generate.go
View File

@@ -621,6 +621,16 @@ func (f *Function) JumpIfLessThan(a, b Register, label string) {
f.AddTrailingComment("jump to:", label, "if", a, "<", b) f.AddTrailingComment("jump to:", label, "if", a, "<", b)
} }
func (f *Function) JumpIfLessThanOrEqual(a, b Register, label string) {
f.compare(a, b)
if f.ISA.Goarch == ARM64 {
f.instr("BLE", label)
} else {
f.instr("JLE", label)
}
f.AddTrailingComment("jump to:", label, "if", a, "<=", b)
}
func (f *Function) JumpIfEqual(a, b Register, label string) { func (f *Function) JumpIfEqual(a, b Register, label string) {
f.compare(a, b) f.compare(a, b)
if f.ISA.Goarch == ARM64 { if f.ISA.Goarch == ARM64 {
@@ -1350,6 +1360,7 @@ func (s *State) index_func(f *Function, test_bytes func(bytes_to_test, test_ans
bytes_to_test := f.Vec() bytes_to_test := f.Vec()
data_start := f.LoadParam(`data`) data_start := f.LoadParam(`data`)
limit := f.LoadParamLen(`data`) limit := f.LoadParamLen(`data`)
f.JumpIfZero(limit, "fail")
f.AddToSelf(limit, data_start) f.AddToSelf(limit, data_start)
mask := f.Reg() mask := f.Reg()
@@ -1357,30 +1368,24 @@ func (s *State) index_func(f *Function, test_bytes func(bytes_to_test, test_ans
f.CopyRegister(data_start, pos) f.CopyRegister(data_start, pos)
func() { func() {
extra_bytes := f.RegForShifts() unaligned_bytes := f.RegForShifts()
defer f.ReleaseReg(extra_bytes) defer f.ReleaseReg(unaligned_bytes)
f.CopyRegister(data_start, extra_bytes) f.CopyRegister(data_start, unaligned_bytes)
f.AndSelf(extra_bytes, vecsz-1) f.AndSelf(unaligned_bytes, vecsz-1)
f.SubtractFromSelf(pos, extra_bytes) f.SubtractFromSelf(pos, unaligned_bytes)
f.Comment(fmt.Sprintf("%s is now aligned to a %d byte boundary so loading from it is safe", pos, vecsz)) f.Comment(fmt.Sprintf("%s is now aligned to a %d byte boundary so loading from it is safe", pos, vecsz))
f.LoadPointerAligned(pos, bytes_to_test) f.LoadPointerAligned(pos, bytes_to_test)
test_bytes(bytes_to_test, test_ans) test_bytes(bytes_to_test, test_ans)
f.MaskForCountDestructive(test_ans, mask) f.MaskForCountDestructive(test_ans, mask)
f.Comment("We need to shift out the possible extra bytes at the start of the string caused by the unaligned read") f.Comment("We need to shift out the possible extra bytes at the start of the string caused by the unaligned read")
f.ShiftMaskRightDestructive(mask, extra_bytes) f.ShiftMaskRightDestructive(mask, unaligned_bytes)
f.JumpIfNonZero(mask, "byte_found_in_mask") f.JumpIfNonZero(mask, "byte_found_in_mask")
}() }()
f.Comment("Now loop over aligned blocks") f.Comment("Now loop over aligned blocks")
fail := func() {
f.SetReturnValue("ans", -1)
f.Return()
}
f.Label("loop_start") f.Label("loop_start")
f.AddToSelf(pos, vecsz) f.AddToSelf(pos, vecsz)
f.JumpIfLessThan(pos, limit, "loop_body") f.JumpIfLessThanOrEqual(limit, pos, "fail")
fail()
f.Label("loop_body")
f.LoadPointerAligned(pos, bytes_to_test) f.LoadPointerAligned(pos, bytes_to_test)
test_bytes(bytes_to_test, test_ans) test_bytes(bytes_to_test, test_ans)
f.JumpIfNonZero(test_ans, "byte_found_in_vec") f.JumpIfNonZero(test_ans, "byte_found_in_vec")
@@ -1392,13 +1397,13 @@ func (s *State) index_func(f *Function, test_bytes func(bytes_to_test, test_ans
f.Label("byte_found_in_mask") f.Label("byte_found_in_mask")
f.CountLeadingZeroBytesInMask(mask, mask) f.CountLeadingZeroBytesInMask(mask, mask)
f.AddToSelf(mask, pos) f.AddToSelf(mask, pos)
f.JumpIfLessThan(mask, limit, "result_in_bounds") f.JumpIfLessThanOrEqual(limit, mask, "fail")
fail()
f.Label("result_in_bounds")
f.SubtractFromSelf(mask, data_start) f.SubtractFromSelf(mask, data_start)
f.SetReturnValue("ans", mask) f.SetReturnValue("ans", mask)
f.Return() f.Return()
f.Label("fail")
f.SetReturnValue("ans", -1)
f.Return()
} }
func (s *State) indexbyte2_body(f *Function) { func (s *State) indexbyte2_body(f *Function) {