From 64bb76b3d970c20b8b195cdb02e0fd2638f6add4 Mon Sep 17 00:00:00 2001
From: Sun Yimin <emmansun@users.noreply.github.com>
Date: Fri, 23 Aug 2024 17:49:51 +0800
Subject: [PATCH] internal/sm2ec: try VLEF, fix issue

---
 internal/sm2ec/p256_asm_s390x.s | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/internal/sm2ec/p256_asm_s390x.s b/internal/sm2ec/p256_asm_s390x.s
index 400be4b..48948ad 100644
--- a/internal/sm2ec/p256_asm_s390x.s
+++ b/internal/sm2ec/p256_asm_s390x.s
@@ -929,6 +929,7 @@ TEXT sm2p256OrdSqrInternal<>(SB), NOFRAME|NOSPLIT, $0
 #define M1    V5
 #define T0    V6
 #define T1    V7
+#define K0    V31
 TEXT ·p256OrdMul(SB), NOSPLIT, $0
 	MOVD res+0(FP), res_ptr
 	MOVD in1+8(FP), x_ptr
@@ -973,6 +974,7 @@ TEXT ·p256OrdMul(SB), NOSPLIT, $0
 #undef M1
 #undef T0
 #undef T1
+#undef K0
 
 // ---------------------------------------
 //  func p256OrdSqr(res, in *p256OrdElement, n int)
@@ -986,6 +988,7 @@ TEXT ·p256OrdMul(SB), NOSPLIT, $0
 #define M1    V5
 #define T0    V6
 #define T1    V7
+#define K0    V31
 TEXT ·p256OrdSqr(SB), NOSPLIT, $0
 	MOVD res+0(FP), res_ptr
 	MOVD in+8(FP), x_ptr
@@ -1033,6 +1036,7 @@ loop:
 #undef M1
 #undef T0
 #undef T1
+#undef K0
 
 TEXT ·p256Mul(SB), NOSPLIT, $0
 	RET