summaryrefslogtreecommitdiff
path: root/scripts/generator
diff options
context:
space:
mode:
Diffstat (limited to 'scripts/generator')
-rwxr-xr-xscripts/generator/generator.py2
-rw-r--r--scripts/generator/generator/datatype.py4
2 files changed, 3 insertions, 3 deletions
diff --git a/scripts/generator/generator.py b/scripts/generator/generator.py
index f7ef4528..8c13b2ff 100755
--- a/scripts/generator/generator.py
+++ b/scripts/generator/generator.py
@@ -42,7 +42,7 @@ FILES = [
"/include/clblast_netlib_c.h",
"/src/clblast_netlib_c.cpp",
]
-HEADER_LINES = [122, 77, 126, 23, 29, 41, 29, 65, 32]
+HEADER_LINES = [122, 77, 126, 24, 29, 41, 29, 65, 32]
FOOTER_LINES = [25, 139, 27, 38, 6, 6, 6, 9, 2]
HEADER_LINES_DOC = 0
FOOTER_LINES_DOC = 63
diff --git a/scripts/generator/generator/datatype.py b/scripts/generator/generator/datatype.py
index 6ac5681a..fdb584bc 100644
--- a/scripts/generator/generator/datatype.py
+++ b/scripts/generator/generator/datatype.py
@@ -72,8 +72,8 @@ class DataType:
def test_template(self):
"""Returns the template as used in the correctness/performance tests"""
- buffer_type = "clblast::" + self.buffer_type if self.buffer_type in [D_FLOAT2, D_DOUBLE2] else self.buffer_type
- beta_cpp = "clblast::" + self.beta_cpp if self.beta_cpp in [D_FLOAT2, D_DOUBLE2] else self.beta_cpp
+ buffer_type = "clblast::" + self.buffer_type if self.is_non_standard() else self.buffer_type
+ beta_cpp = "clblast::" + self.beta_cpp if self.beta_cpp in [D_HALF, D_FLOAT2, D_DOUBLE2] else self.beta_cpp
if self.buffer_type != self.beta_cpp:
return "<" + buffer_type + "," + self.beta_cpp + ">, " + buffer_type + ", " + beta_cpp
return "<" + buffer_type + ">, " + buffer_type + ", " + beta_cpp