shuren/opencl_fdfd
1
0
Fork 0
forked from jan/opencl_fdfd
Code Issues Pull requests Releases Wiki Activity

add half-precision to type_to_C

Browse source
This commit is contained in:
jan 2022-11-24 22:33:10 -08:00
commit c42ce49983
1 changed files with 1 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
opencl_fdfd/ops.py
View file

@ -42,6 +42,7 @@ def type_to_C(
string containing the corresponding C type (eg. 'double') string containing the corresponding C type (eg. 'double')
""" """
types = { types = {
numpy.float16: 'half',
numpy.float32: 'float', numpy.float32: 'float',
numpy.float64: 'double', numpy.float64: 'double',
numpy.complex64: 'cfloat_t', numpy.complex64: 'cfloat_t',