go/src/internal/bytealg/count_riscv64.s
Joel Sing 6fc1e34100 internal/bytealg: deduplicate code between Count/CountString for riscv64
Change-Id: I22eb4e7444e5fe5f6767cc960895f3c6e2fa13cc
Reviewed-on: https://go-review.googlesource.com/c/go/+/661615
Reviewed-by: Keith Randall <khr@google.com>
LUCI-TryBot-Result: Go LUCI <golang-scoped@luci-project-accounts.iam.gserviceaccount.com>
Reviewed-by: Keith Randall <khr@golang.org>
Auto-Submit: Carlos Amedee <carlos@golang.org>
Reviewed-by: Carlos Amedee <carlos@golang.org>
2025-05-01 05:56:26 -07:00

36 lines
701 B
ArmAsm

// Copyright 2020 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.
#include "go_asm.h"
#include "textflag.h"
TEXT ·CountString<ABIInternal>(SB),NOSPLIT,$0-32
// X10 = s_base
// X11 = s_len
// X12 = byte to count
MOV X12, X13
JMP ·Count<ABIInternal>(SB)
TEXT ·Count<ABIInternal>(SB),NOSPLIT,$0-40
// X10 = b_base
// X11 = b_len
// X12 = b_cap (unused)
// X13 = byte to count (want in X12)
AND $0xff, X13, X12
MOV ZERO, X14 // count
ADD X10, X11 // end
PCALIGN $16
loop:
BEQ X10, X11, done
MOVBU (X10), X15
ADD $1, X10
BNE X12, X15, loop
ADD $1, X14
JMP loop
done:
MOV X14, X10
RET