From 6938758f6c1754f0ee49d0709dd0ca376a146010 Mon Sep 17 00:00:00 2001
From: Adrian Kummerlaender
Date: Sun, 27 Oct 2019 16:03:07 +0100
Subject: Move layout implementations into separate folder

---
 boltzgen/kernel/generator.py | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

(limited to 'boltzgen/kernel/generator.py')

diff --git a/boltzgen/kernel/generator.py b/boltzgen/kernel/generator.py
index 04b0a64..ebe1018 100644
--- a/boltzgen/kernel/generator.py
+++ b/boltzgen/kernel/generator.py
@@ -3,8 +3,7 @@ import sympy
 from mako.template import Template
 from pathlib import Path
 
-import kernel.target.cl
-import kernel.target.cpp
+import kernel.target.layout
 
 class Generator:
     def __init__(self, descriptor, moments, collision):
@@ -33,8 +32,8 @@ class Generator:
             extras = extras
         )
 
-    def kernel(self, target, precision, layout, geometry, functions = ['collide_and_stream'], extras = []):
-        layout_impl = eval("kernel.target.%s.%s" % (target, layout))
+    def kernel(self, target, precision, layout, geometry, functions, extras = []):
+        layout_impl = eval("kernel.target.layout.%s.%s" % (target, layout))
         if layout_impl is None:
             raise Exception("Target '%s' doesn't support layout '%s'" % (target, layout))
         else:
-- 
cgit v1.2.3