From 95e91b44d2ef3535129c0a402c51bc56cfd74d06 Mon Sep 17 00:00:00 2001
From: Josh Blum
Date: Sun, 15 Apr 2012 17:32:38 -0700
Subject: volk: created other templates for runtime + machines

---
 volk/gen/volk_kernel_defs.py | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

(limited to 'volk/gen/volk_kernel_defs.py')

diff --git a/volk/gen/volk_kernel_defs.py b/volk/gen/volk_kernel_defs.py
index f598ed3c0..a5c7cb710 100644
--- a/volk/gen/volk_kernel_defs.py
+++ b/volk/gen/volk_kernel_defs.py
@@ -154,8 +154,7 @@ for func in functions:
     ret = replace.sub("", ret);
     replace = re.compile("inline ");
     ret = replace.sub("", ret);
-    replace = re.compile("\)");
-    arched_args = replace.sub(", const char* arch) {", args);
+    arched_args = args[args.find('(')+1:args.find(')')]
 
     remove = re.compile('\)|\(|{');
     rargs = remove.sub("", args);
@@ -198,8 +197,13 @@ for func in functions:
 class kernel_class:
     def __init__(self, index):
         self.name = functions[index]
+        self.pname = self.name.replace('volk_', 'p_')
         self.rettype = retlist[index]
-        self.arglist = my_argtypelist[index]
+        self.arglist_defs = my_argtypelist[index]
+        self.arglist_namedefs = arched_arglist[index]
+        self.arglist_names = my_arglist[index]
+        self.tagdeps = fcountlist[index]
+        self.taglist = taglist[index]
 
     def __repr__(self):
         return self.name
-- 
cgit