BLD: Re-work the L2 gemv

This commit is contained in:
Rohit Goswami 2024-04-28 15:36:26 +00:00 committed by Mateusz Sokół
parent 69edd1d5db
commit 7587dc9975
1 changed files with 60 additions and 58 deletions

View File

@ -182,64 +182,66 @@ base_kops = [
# exts are used to find the flags for each modality # exts are used to find the flags for each modality
# ext is attached to base (only useful from Level 2) # ext is attached to base (only useful from Level 2)
# i.e. baseext (e.g., gemv_n, gemv_t, cgeru_k, cgerc_k) # i.e. baseext (e.g., gemv_n, gemv_t, cgeru_k, cgerc_k)
# {'base': '?gemv', {'base': '?gemv',
# 'modes': { 'modes': {
# 's': { 's': {
# 'exts': { 'exts': {
# '_n': {'dir': 'arm', 'kernel': 'gemv_n.c'}, # TODO(rg): Where are these coming from??
# '_t': {'dir': 'arm', 'kernel': 'gemv_n.c'}, # Makefile lists sgemv_n_4.c as the source, though there is a sgemv_n.c
# } '_n': {'dir': 'x86_64', 'kernel': 'sgemv_n_4.c'},
# }, '_t': {'dir': 'x86_64', 'kernel': 'sgemv_t_4.c'},
# 'd': { }
# 'exts': { },
# '_n': {'dir': 'arm', 'kernel': 'gemv_n.c'}, 'd': {
# '_t': {'dir': 'arm', 'kernel': 'gemv_n.c'}, 'exts': {
# } '_n': {'dir': 'arm', 'kernel': 'gemv_n.c'},
# }, '_t': {'dir': 'arm', 'kernel': 'gemv_t.c'},
# 'q': { }
# 'exts': { },
# '_n': {'dir': 'arm', 'kernel': 'gemv_n.c'}, 'q': {
# '_t': {'dir': 'arm', 'kernel': 'gemv_n.c'}, 'exts': {
# } '_n': {'dir': 'arm', 'kernel': 'gemv_n.c'},
# }, '_t': {'dir': 'arm', 'kernel': 'gemv_n.c'},
# 'c': { }
# 'exts': { },
# '_n': {'dir': 'arm', 'kernel': 'zgemv_n.c'}, 'c': {
# '_t': {'dir': 'arm', 'kernel': 'zgemv_n.c'}, 'exts': {
# '_r': {'dir': 'arm', 'kernel': 'zgemv_n.c'}, '_n': {'dir': 'arm', 'kernel': 'zgemv_n.c'},
# '_c': {'dir': 'arm', 'kernel': 'zgemv_n.c'}, '_t': {'dir': 'arm', 'kernel': 'zgemv_n.c'},
# '_o': {'dir': 'arm', 'kernel': 'zgemv_n.c'}, '_r': {'dir': 'arm', 'kernel': 'zgemv_n.c'},
# '_u': {'dir': 'arm', 'kernel': 'zgemv_n.c'}, '_c': {'dir': 'arm', 'kernel': 'zgemv_n.c'},
# '_s': {'dir': 'arm', 'kernel': 'zgemv_n.c'}, '_o': {'dir': 'arm', 'kernel': 'zgemv_n.c'},
# '_d': {'dir': 'arm', 'kernel': 'zgemv_n.c'}, '_u': {'dir': 'arm', 'kernel': 'zgemv_n.c'},
# } '_s': {'dir': 'arm', 'kernel': 'zgemv_n.c'},
# }, '_d': {'dir': 'arm', 'kernel': 'zgemv_n.c'},
# 'z': { }
# 'exts': { },
# '_n': {'dir': 'arm', 'kernel': 'zgemv_n.c'}, 'z': {
# '_t': {'dir': 'arm', 'kernel': 'zgemv_n.c'}, 'exts': {
# '_r': {'dir': 'arm', 'kernel': 'zgemv_n.c'}, '_n': {'dir': 'arm', 'kernel': 'zgemv_n.c'},
# '_c': {'dir': 'arm', 'kernel': 'zgemv_n.c'}, '_t': {'dir': 'arm', 'kernel': 'zgemv_n.c'},
# '_o': {'dir': 'arm', 'kernel': 'zgemv_n.c'}, '_r': {'dir': 'arm', 'kernel': 'zgemv_n.c'},
# '_u': {'dir': 'arm', 'kernel': 'zgemv_n.c'}, '_c': {'dir': 'arm', 'kernel': 'zgemv_n.c'},
# '_s': {'dir': 'arm', 'kernel': 'zgemv_n.c'}, '_o': {'dir': 'arm', 'kernel': 'zgemv_n.c'},
# '_d': {'dir': 'arm', 'kernel': 'zgemv_n.c'}, '_u': {'dir': 'arm', 'kernel': 'zgemv_n.c'},
# } '_s': {'dir': 'arm', 'kernel': 'zgemv_n.c'},
# }, '_d': {'dir': 'arm', 'kernel': 'zgemv_n.c'},
# 'x': { }
# 'exts': { },
# '_n': {'dir': 'arm', 'kernel': 'zgemv_n.c'}, 'x': {
# '_t': {'dir': 'arm', 'kernel': 'zgemv_n.c'}, 'exts': {
# '_r': {'dir': 'arm', 'kernel': 'zgemv_n.c'}, '_n': {'dir': 'arm', 'kernel': 'zgemv_n.c'},
# '_c': {'dir': 'arm', 'kernel': 'zgemv_n.c'}, '_t': {'dir': 'arm', 'kernel': 'zgemv_n.c'},
# '_o': {'dir': 'arm', 'kernel': 'zgemv_n.c'}, '_r': {'dir': 'arm', 'kernel': 'zgemv_n.c'},
# '_u': {'dir': 'arm', 'kernel': 'zgemv_n.c'}, '_c': {'dir': 'arm', 'kernel': 'zgemv_n.c'},
# '_s': {'dir': 'arm', 'kernel': 'zgemv_n.c'}, '_o': {'dir': 'arm', 'kernel': 'zgemv_n.c'},
# '_d': {'dir': 'arm', 'kernel': 'zgemv_n.c'}, '_u': {'dir': 'arm', 'kernel': 'zgemv_n.c'},
# } '_s': {'dir': 'arm', 'kernel': 'zgemv_n.c'},
# }, '_d': {'dir': 'arm', 'kernel': 'zgemv_n.c'},
# }, }
# }, },
},
},
# {'base': '?symv', # {'base': '?symv',
# 'modes': { # 'modes': {
# 's': { # 's': {