Skip to content

Commit 70294c3

Browse files
committed
Accommodate ex and quad precision builds.
1 parent 2d62a22 commit 70294c3

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

common_macro.h

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -149,6 +149,11 @@
149149
#define GEMM_KERNEL_R QGEMM_KERNEL
150150
#define GEMM_KERNEL_B QGEMM_KERNEL
151151

152+
#define COMM_KERNEL_N QGEMM_KERNEL
153+
#define COMM_KERNEL_L QGEMM_KERNEL
154+
#define COMM_KERNEL_R QGEMM_KERNEL
155+
#define COMM_KERNEL_B QGEMM_KERNEL
156+
152157
#define TRMM_KERNEL_LN QTRMM_KERNEL_LN
153158
#define TRMM_KERNEL_LT QTRMM_KERNEL_LT
154159
#define TRMM_KERNEL_LR QTRMM_KERNEL_LN
@@ -1541,6 +1546,11 @@
15411546
#define GEMM_KERNEL_R XGEMM_KERNEL_R
15421547
#define GEMM_KERNEL_B XGEMM_KERNEL_B
15431548

1549+
#define COMM_KERNEL_N XGEMM_KERNEL_N
1550+
#define COMM_KERNEL_L XGEMM_KERNEL_L
1551+
#define COMM_KERNEL_R XGEMM_KERNEL_R
1552+
#define COMM_KERNEL_B XGEMM_KERNEL_B
1553+
15441554
#define GEMM3M_KERNEL XGEMM3M_KERNEL
15451555

15461556
#define TRMM_KERNEL_LN XTRMM_KERNEL_LN

0 commit comments

Comments
 (0)