MAINT: Fixup and use the new .S dep

This commit is contained in:
Rohit Goswami 2024-07-01 02:18:03 +00:00 committed by Mateusz Sokół
parent 17b164feee
commit 3094d0c621
1 changed files with 8 additions and 8 deletions

View File

@ -429,8 +429,8 @@ base_kops = [
# Level 3 symbols # Level 3 symbols
{ 'base': '?gemm_kernel', { 'base': '?gemm_kernel',
'modes': { 'modes': {
's': {'exts': {'': {'dir': 'generic', 'kernel': 'gemmkernel_2x2.c'}}}, 's': {'exts': {'': {'dir': 'x86_64', 'kernel': 'sgemm_kernel_8x4_haswell.c'}}},
'd': {'exts': {'': {'dir': 'generic', 'kernel': 'gemmkernel_2x2.c'}}}, 'd': {'exts': {'': {'dir': 'x86_64', 'kernel': 'dgemm_kernel_4x8_haswell.S'}}},
'c': { 'c': {
'exts': { 'exts': {
'_n': {'dir': 'generic', 'kernel': 'zgemmkernel_2x2.c', 'addl': ['-DNN']}, '_n': {'dir': 'generic', 'kernel': 'zgemmkernel_2x2.c', 'addl': ['-DNN']},
@ -456,10 +456,10 @@ base_kops = [
'modes': { 'modes': {
's': { 's': {
'exts': { 'exts': {
'_LN': {'dir': 'generic', 'kernel': 'trmmkernel_2x2.c'}, '_LN': {'dir': 'x86_64', 'kernel': 'sgemm_kernel_8x4_haswell.c'},
'_LT': {'dir': 'generic', 'kernel': 'trmmkernel_2x2.c', 'addl': ['-DLEFT', '-DTRANSA']}, '_LT': {'dir': 'x86_64', 'kernel': 'sgemm_kernel_8x4_haswell.c', 'addl': ['-DLEFT', '-DTRANSA']},
'_RN': {'dir': 'generic', 'kernel': 'trmmkernel_2x2.c'}, '_RN': {'dir': 'x86_64', 'kernel': 'sgemm_kernel_8x4_haswell.c'},
'_RT': {'dir': 'generic', 'kernel': 'trmmkernel_2x2.c'}, '_RT': {'dir': 'x86_64', 'kernel': 'sgemm_kernel_8x4_haswell.c'},
} }
}, },
'd': { 'd': {
@ -589,7 +589,7 @@ base_kops = [
# TODO(rg): the _NUM prefixes are arch dependent # TODO(rg): the _NUM prefixes are arch dependent
'_incopy': {'dir': 'generic', 'kernel': 'gemm_ncopy_8.c'}, '_incopy': {'dir': 'generic', 'kernel': 'gemm_ncopy_8.c'},
'_itcopy': {'dir': 'generic', 'kernel': 'gemm_tcopy_8.c'}, '_itcopy': {'dir': 'generic', 'kernel': 'gemm_tcopy_8.c'},
'_oncopy': {'dir': 'generic', 'kernel': 'gemm_ncopy_2.c'}, '_oncopy': {'dir': 'generic', 'kernel': 'gemm_ncopy_4.c'},
'_otcopy': {'dir': 'generic', 'kernel': 'gemm_tcopy_4.c'}, '_otcopy': {'dir': 'generic', 'kernel': 'gemm_tcopy_4.c'},
# TODO(rg): direct and direct_performant are built only conditionally # TODO(rg): direct and direct_performant are built only conditionally
'_direct': {'dir': 'x86_64', 'kernel': 'sgemm_direct_skylakex.c'}, '_direct': {'dir': 'x86_64', 'kernel': 'sgemm_direct_skylakex.c'},
@ -601,7 +601,7 @@ base_kops = [
'kernel': 'gemm_small_matrix_permit.c'}, 'kernel': 'gemm_small_matrix_permit.c'},
'_incopy': {'dir': 'generic', 'kernel': 'gemm_ncopy_8.c'}, '_incopy': {'dir': 'generic', 'kernel': 'gemm_ncopy_8.c'},
'_itcopy': {'dir': 'generic', 'kernel': 'gemm_tcopy_8.c'}, '_itcopy': {'dir': 'generic', 'kernel': 'gemm_tcopy_8.c'},
'_oncopy': {'dir': 'generic', 'kernel': 'gemm_ncopy_2.c'}, '_oncopy': {'dir': 'generic', 'kernel': 'gemm_ncopy_4.c'},
'_otcopy': {'dir': 'generic', 'kernel': 'gemm_tcopy_4.c'}, '_otcopy': {'dir': 'generic', 'kernel': 'gemm_tcopy_4.c'},
}}, }},
'c': {'exts': { 'c': {'exts': {