fix sve ztrmm kernel
This commit is contained in:
parent
68c414d3a6
commit
2e2c02b762
|
@ -723,7 +723,7 @@ USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||||
add pB, pB, temp
|
add pB, pB, temp
|
||||||
#endif
|
#endif
|
||||||
#if defined(LEFT)
|
#if defined(LEFT)
|
||||||
add tempOffset, tempOffset, #4
|
add tempOffset, tempOffset, lanes
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
prfm PLDL1KEEP, [pA]
|
prfm PLDL1KEEP, [pA]
|
||||||
|
@ -856,7 +856,7 @@ USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||||
add pB, pB, temp
|
add pB, pB, temp
|
||||||
#endif
|
#endif
|
||||||
#if defined(LEFT)
|
#if defined(LEFT)
|
||||||
add tempOffset, tempOffset, #4
|
add tempOffset, tempOffset, lanes
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
.Lztrmm_kernel_L2_Mv1_END:
|
.Lztrmm_kernel_L2_Mv1_END:
|
||||||
|
@ -923,7 +923,7 @@ USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||||
add tempK, tempOffset, #1
|
add tempK, tempOffset, #1
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
asr counterL , temp, #3 // counterL = counterL / 8
|
asr counterL , tempK, #3 // counterL = counterL / 8
|
||||||
cmp counterL , #0
|
cmp counterL , #0
|
||||||
ble .Lztrmm_kernel_L1_Mv1_40
|
ble .Lztrmm_kernel_L1_Mv1_40
|
||||||
.align 5
|
.align 5
|
||||||
|
@ -972,7 +972,7 @@ USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||||
add pB, pB, temp
|
add pB, pB, temp
|
||||||
#endif
|
#endif
|
||||||
#if defined(LEFT)
|
#if defined(LEFT)
|
||||||
add tempOffset, tempOffset, #4
|
add tempOffset, tempOffset, lanes
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
.Lztrmm_kernel_L1_Mv1_END:
|
.Lztrmm_kernel_L1_Mv1_END:
|
||||||
|
|
Loading…
Reference in New Issue