Floating point conversion in Cython wrapper.

scripts/generator/generator.py

@@ -49,7 +49,7 @@
     "/src/clblast_cuda.cpp",
     "/src/pyclblast/src/pyclblast.pyx"
 ]
-HEADER_LINES = [129, 21, 133, 24, 29, 45, 29, 66, 40, 96, 21, 327]
+HEADER_LINES = [129, 21, 133, 24, 29, 45, 29, 66, 40, 96, 21, 341]
 FOOTER_LINES = [98, 57, 112, 275, 6, 6, 6, 9, 2, 41, 56, 37]
 HEADER_LINES_DOC = 0
 FOOTER_LINES_DOC = 232
@@ -215,7 +215,7 @@ def main(argv):
         file_footer = original[-FOOTER_LINES[i]:]
 
         # Re-writes the body of the file
-        with open(library_root + FILES[i], "w") as f:
+        with open(library_root + FILES[i], "w", newline="\n") as f:
             body = ""
             levels = [1, 2, 3] if (i == 4 or i == 5 or i == 6) else [1, 2, 3, 4]
             for level in levels:
@@ -261,14 +261,14 @@ def main(argv):
 
                 # Correctness tests
                 filename = library_root + "/test/correctness/routines/" + routine_suffix
-                with open(filename, "w") as f:
+                with open(filename, "w", newline="\n") as f:
                     f.write(cpp.HEADER + "\n")
                     f.write(cpp.correctness_test(routine, level_string))
                     f.write(cpp.FOOTER)
 
                 # Performance tests
                 filename = library_root + "/test/performance/routines/" + routine_suffix
-                with open(filename, "w") as f:
+                with open(filename, "w", newline="\n") as f:
                     f.write(cpp.HEADER + "\n")
                     f.write(cpp.performance_test(routine, level_string))
                     f.write(cpp.FOOTER)
@@ -283,7 +283,7 @@ def main(argv):
     file_footer = original[-FOOTER_LINES_DOC:]
 
     # Outputs the API documentation
-    with open(filename, "w") as f:
+    with open(filename, "w", newline="\n") as f:
 
         # Outputs the header
         f.write("".join(file_header))

scripts/generator/generator/pyclblast.py

@@ -8,7 +8,7 @@
 import os
 
 
-NL = os.linesep
+NL = '\n'
 SEPARATOR = "####################################################################################################"
 
 
@@ -43,7 +43,7 @@ def scalar_cython_conversion(scalar, flavour):
     if scalar_type in ["cl_double2", "double2"]:
         return "<cl_double2>cl_double2(x=" + scalar + ".real,y=" + scalar + ".imag)"
     if scalar_type in ["cl_half", "half"]:
-        return "<cl_half>" + scalar
+        return "<cl_half>val_to_half(" + scalar + ")"
     raise RuntimeError("Could not convert flavour '%s:%s'" % (flavour.precision_name, scalar_type))
 
 
@@ -82,8 +82,18 @@ def generate_pyx(routine):
         result += NL
 
         # Data types and checks
-        result += indent + "dtype = check_dtype([" + ", ".join(buffers) + "], "
+        int_buff = []
+        other_buff = []
+        for buf in buffers:
+            if buf in routine.index_buffers():
+                int_buff.append(buf)
+            else:
+                other_buff.append(buf)
+        result += indent + "dtype = check_dtype([" + ", ".join(other_buff) + "], "
         result += "[" + ", ".join(['"%s"' % d for d in np_dtypes]) + "])" + NL
+        if int_buff:
+            result += indent + "check_dtype([" + ", ".join(int_buff) + "], "
+            result += "[" + ", ".join(['"uint16", "uint32", "uint64"']) + "])" + NL
         for buf in buffers:
             if buf in routine.buffers_vector():
                 result += indent + "check_vector("

src/pyclblast/setup.py

@@ -10,6 +10,11 @@
 from distutils.extension import Extension
 from Cython.Distutils import build_ext
 import platform
+import numpy
+import os
+
+np_incdir = numpy.get_include()
+np_libdir = os.path.join(np_incdir, '..', 'lib', '')
 
 runtime_library_dirs = list()
 if platform.system() == "Linux":
@@ -23,8 +28,10 @@
     Extension(
         "pyclblast",
         ["src/pyclblast.pyx"],
-        libraries=["clblast"],
+        libraries=["clblast", "npymath"],
         runtime_library_dirs=runtime_library_dirs,
+        library_dirs=[np_libdir],
+        include_dirs=[np_incdir],
         language="c++"
     )
 )