We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 651578e commit 959d3b3Copy full SHA for 959d3b3
1 file changed
common_macro.h
@@ -149,6 +149,11 @@
149
#define GEMM_KERNEL_R QGEMM_KERNEL
150
#define GEMM_KERNEL_B QGEMM_KERNEL
151
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
+
157
#define TRMM_KERNEL_LN QTRMM_KERNEL_LN
158
#define TRMM_KERNEL_LT QTRMM_KERNEL_LT
159
#define TRMM_KERNEL_LR QTRMM_KERNEL_LN
@@ -1541,6 +1546,11 @@
1541
1546
#define GEMM_KERNEL_R XGEMM_KERNEL_R
1542
1547
#define GEMM_KERNEL_B XGEMM_KERNEL_B
1543
1548
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
1544
1554
#define GEMM3M_KERNEL XGEMM3M_KERNEL
1545
1555
1556
#define TRMM_KERNEL_LN XTRMM_KERNEL_LN
0 commit comments