Skip to content

Commit b7b408a

Browse files
authored
optimize AVX2 SGEMM
1 parent 92b1021 commit b7b408a

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

param.h

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -625,7 +625,7 @@ USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
625625

626626
#else
627627

628-
#define SGEMM_DEFAULT_UNROLL_M 16
628+
#define SGEMM_DEFAULT_UNROLL_M 8
629629
#define DGEMM_DEFAULT_UNROLL_M 4
630630
#define QGEMM_DEFAULT_UNROLL_M 2
631631
#define CGEMM_DEFAULT_UNROLL_M 8
@@ -666,7 +666,7 @@ USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
666666

667667
#else
668668

669-
#define SGEMM_DEFAULT_P 768
669+
#define SGEMM_DEFAULT_P 320
670670
#define DGEMM_DEFAULT_P 512
671671
#define CGEMM_DEFAULT_P 256
672672
#define ZGEMM_DEFAULT_P 192
@@ -675,7 +675,7 @@ USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
675675
#define SGEMM_DEFAULT_Q 320
676676
#define DGEMM_DEFAULT_Q 128
677677
#else
678-
#define SGEMM_DEFAULT_Q 384
678+
#define SGEMM_DEFAULT_Q 320
679679
#define DGEMM_DEFAULT_Q 256
680680
#endif
681681
#define CGEMM_DEFAULT_Q 256
@@ -1528,7 +1528,7 @@ USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
15281528

15291529
#else
15301530

1531-
#define SGEMM_DEFAULT_UNROLL_M 16
1531+
#define SGEMM_DEFAULT_UNROLL_M 8
15321532
#define DGEMM_DEFAULT_UNROLL_M 4
15331533
#define QGEMM_DEFAULT_UNROLL_M 2
15341534
#define CGEMM_DEFAULT_UNROLL_M 8
@@ -1569,7 +1569,7 @@ USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
15691569

15701570
#else
15711571

1572-
#define SGEMM_DEFAULT_P 768
1572+
#define SGEMM_DEFAULT_P 320
15731573
#define DGEMM_DEFAULT_P 512
15741574
#define CGEMM_DEFAULT_P 256
15751575
#define ZGEMM_DEFAULT_P 192
@@ -1578,7 +1578,7 @@ USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
15781578
#define SGEMM_DEFAULT_Q 320
15791579
#define DGEMM_DEFAULT_Q 128
15801580
#else
1581-
#define SGEMM_DEFAULT_Q 384
1581+
#define SGEMM_DEFAULT_Q 320
15821582
#define DGEMM_DEFAULT_Q 256
15831583
#endif
15841584
#define CGEMM_DEFAULT_Q 256

0 commit comments

Comments
 (0)