ENH: Start abstracting rules for kernels
This commit is contained in:
parent
0043edb066
commit
e9a3897174
|
@ -0,0 +1,98 @@
|
||||||
|
# Ordered As per https://netlib.org/blas/blasqr.pdf
|
||||||
|
# NOTE: xROTG xROTMG xROTM have no kernels?
|
||||||
|
_kinds = []
|
||||||
|
_rk = ['s', 'd']
|
||||||
|
_ck = ['c', 'z']
|
||||||
|
_std_dk = _rk + _ck # Standard precisions
|
||||||
|
# TODO: Actually test and set this
|
||||||
|
if fma3
|
||||||
|
fma3_flag = '-mfma'
|
||||||
|
endif
|
||||||
|
# TODO: This is currently following x86_64 generic for src and dir, but it needs
|
||||||
|
# to diversify
|
||||||
|
# NOTE: The def and undefs are from Makefile.L1
|
||||||
|
# Construct all PRECsymbKERNEL from src and dir via files(dir + src)
|
||||||
|
# For the precs array, the following mapping is used for c_args:
|
||||||
|
# undef --> -Uwhatever
|
||||||
|
# def --> -Dwhatever
|
||||||
|
# addl --> passed AS IS
|
||||||
|
kops_array = [
|
||||||
|
# Level 1 BLAS
|
||||||
|
{
|
||||||
|
'base': 'rot',
|
||||||
|
# TODO: A lot of these flags are basically repeating..
|
||||||
|
'precs': {
|
||||||
|
's' : {
|
||||||
|
'dir': 'arm',
|
||||||
|
'kernel': 'rot.c',
|
||||||
|
'undef': ['COMPLEX', 'DOUBLE'],
|
||||||
|
'addl': [fma3_flag],
|
||||||
|
},
|
||||||
|
'd' : {
|
||||||
|
'dir': 'arm',
|
||||||
|
'kernel': 'rot.c',
|
||||||
|
'undef': ['COMPLEX'],
|
||||||
|
'def': ['DOUBLE'],
|
||||||
|
'addl': [fma3_flag],
|
||||||
|
},
|
||||||
|
'q' : {
|
||||||
|
'dir': 'arm',
|
||||||
|
'kernel': 'zrot.c',
|
||||||
|
'undef': ['COMPLEX'],
|
||||||
|
'def': ['DXDOUBLE'],
|
||||||
|
},
|
||||||
|
'cs' : {
|
||||||
|
'dir': 'arm',
|
||||||
|
'kernel': 'zrot.c',
|
||||||
|
'undef': ['DOUBLE'],
|
||||||
|
'def': ['COMPLEX'],
|
||||||
|
},
|
||||||
|
'zd' : {
|
||||||
|
'dir': 'arm',
|
||||||
|
'kernel': 'zrot.c',
|
||||||
|
'def': ['COMPLEX', 'DOUBLE'],
|
||||||
|
},
|
||||||
|
'xq' : {
|
||||||
|
'dir': 'arm',
|
||||||
|
'kernel': 'zrot.c',
|
||||||
|
'def': ['COMPLEX', 'DXDOUBLE'],
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{'symb': 'swap',
|
||||||
|
'src': {'real': 'swap', 'cmplx': 'zswap', 'ext': '.c'},
|
||||||
|
'dir': 'arm', 'precs': _std_dk },
|
||||||
|
{'symb': 'scal',
|
||||||
|
'src': {'real': 'scal', 'cmplx': 'zscal', 'ext': '.c'},
|
||||||
|
'dir': 'arm', 'precs': _std_dk },
|
||||||
|
{'symb': 'copy',
|
||||||
|
'src': {'real': 'copy', 'cmplx': 'zcopy', 'ext': '.c'},
|
||||||
|
'dir': 'arm', 'precs': _std_dk },
|
||||||
|
{'symb': 'axpy',
|
||||||
|
'src': {'real': 'axpy', 'cmplx': 'zaxpy', 'ext': '.c'},
|
||||||
|
'dir': 'arm', 'precs': _std_dk },
|
||||||
|
{'symb': 'dot',
|
||||||
|
'src': {'real': 'dot', 'cmplx': 'zdot', 'ext': '.c'},
|
||||||
|
'dir': 'arm', 'precs': _std_dk },
|
||||||
|
]
|
||||||
|
|
||||||
|
kernel_confs = []
|
||||||
|
|
||||||
|
foreach root : blas1_roots
|
||||||
|
fname = root + '.S'
|
||||||
|
defs = []
|
||||||
|
foreach prec : real_kinds
|
||||||
|
name = prec + fname + '_k'
|
||||||
|
kernel_confs += {'defs': defs, 'name': name, 'src': fname}
|
||||||
|
endforeach
|
||||||
|
endforeach
|
||||||
|
|
||||||
|
_static_libs = []
|
||||||
|
foreach conf: kernel_confs
|
||||||
|
_static_libs += static_library(
|
||||||
|
conf['name'],
|
||||||
|
conf['src'],
|
||||||
|
include_directories: _inc,
|
||||||
|
c_args: conf['defs'],
|
||||||
|
)
|
||||||
|
endforeach
|
Loading…
Reference in New Issue