Skip to content

Commit 995a990

Browse files
authored
Make AVX512 BFLOAT16 kernels conditional on compiler capability
1 parent 1dada6d commit 995a990

File tree

2 files changed

+4
-1
lines changed

2 files changed

+4
-1
lines changed

kernel/x86_64/KERNEL.COOPERLAKE

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
include $(KERNELDIR)/KERNEL.SKYLAKEX
2-
2+
ifneq ($(NO_AVX512BF16), 1)
33
SBGEMM_SMALL_M_PERMIT = sbgemm_small_kernel_permit_cooperlake.c
44
SBGEMM_SMALL_K_NN = sbgemm_small_kernel_nn_cooperlake.c
55
SBGEMM_SMALL_K_B0_NN = sbgemm_small_kernel_nn_cooperlake.c
@@ -20,3 +20,4 @@ SBGEMMINCOPYOBJ = sbgemm_incopy$(TSUFFIX).$(SUFFIX)
2020
SBGEMMITCOPYOBJ = sbgemm_itcopy$(TSUFFIX).$(SUFFIX)
2121
SBGEMMONCOPYOBJ = sbgemm_oncopy$(TSUFFIX).$(SUFFIX)
2222
SBGEMMOTCOPYOBJ = sbgemm_otcopy$(TSUFFIX).$(SUFFIX)
23+
endif

kernel/x86_64/KERNEL.SAPPHIRERAPIDS

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
include $(KERNELDIR)/KERNEL.COOPERLAKE
22

3+
ifneq ($(NO_AVX512BF16), 1)
34
SBGEMM_SMALL_M_PERMIT =
45
SBGEMM_SMALL_K_NN =
56
SBGEMM_SMALL_K_B0_NN =
@@ -20,3 +21,4 @@ SBGEMMINCOPYOBJ = sbgemm_incopy$(TSUFFIX).$(SUFFIX)
2021
SBGEMMITCOPYOBJ = sbgemm_itcopy$(TSUFFIX).$(SUFFIX)
2122
SBGEMMONCOPYOBJ = sbgemm_oncopy$(TSUFFIX).$(SUFFIX)
2223
SBGEMMOTCOPYOBJ = sbgemm_otcopy$(TSUFFIX).$(SUFFIX)
24+
endif

0 commit comments

Comments
 (0)