From a298781b8529be341ae3b65ff368188cdcdb688a Mon Sep 17 00:00:00 2001
From: Kovid Goyal <kovid@kovidgoyal.net>
Date: Thu, 1 Feb 2024 22:55:07 +0530
Subject: [PATCH] DRYer

---
 tools/simdstring/generate.go | 21 ++++-----------------
 1 file changed, 4 insertions(+), 17 deletions(-)

diff --git a/tools/simdstring/generate.go b/tools/simdstring/generate.go
index 1a23a13d7..752c481e4 100644
--- a/tools/simdstring/generate.go
+++ b/tools/simdstring/generate.go
@@ -752,7 +752,6 @@ func (f *Function) Set1Epi8(val any, vec Register) {
 		f.instr("PSHUFB", shuffle_mask, vec)
 		f.ReleaseReg(shuffle_mask)
 	}
-	defer f.Comment()
 
 	switch v := val.(type) {
 	default:
@@ -766,26 +765,12 @@ func (f *Function) Set1Epi8(val any, vec Register) {
 			f.AllOnesRegister(vec)
 			return
 		}
-
-		f.Comment("Set all bytes of", vec, "to", v)
 		r := f.Reg()
 		defer f.ReleaseReg(r)
 		f.SetRegisterTo(r, v)
-
-		if f.ISA.Goarch == ARM64 {
-			f.instr("VMOV", r, vec.ARMFullWidth())
-		} else {
-			switch vec.Size {
-			case 128:
-				do_shuffle_load(r)
-			case 256:
-				temp := f.Vec(128)
-				defer f.ReleaseReg(temp)
-				f.instr("VMOVD", r, temp)
-				f.instr("VPBROADCASTB", temp, vec)
-			}
-		}
+		f.Set1Epi8(r, vec)
 	case Register:
+		f.Comment("Set all bytes of", vec, "to the lowest byte in", v)
 		if v.Size != f.ISA.GeneralPurposeRegisterSize {
 			panic("Can only set1_epi8 from a general purpose register")
 		}
@@ -802,6 +787,7 @@ func (f *Function) Set1Epi8(val any, vec Register) {
 				f.instr("VPBROADCASTB", temp, vec)
 			}
 		}
+		defer f.Comment()
 	case string:
 		f.Comment("Set all bytes of", vec, "to the first byte in", v)
 		if f.ISA.Goarch == ARM64 {
@@ -818,6 +804,7 @@ func (f *Function) Set1Epi8(val any, vec Register) {
 		case 256:
 			f.instr("VPBROADCASTB", f.ParamPos(v), vec)
 		}
+		defer f.Comment()
 	}
 }