1 files changed, 10 insertions, 3 deletions
diff --git a/track.py b/track.py
index 6f53c09..101e08e 100644
--- a/track.py
+++ b/track.py
@@ -7,8 +7,9 @@ from threadtest import create_job
 import matplotlib.pyplot as plt
 from itertoolset import roundrobin
 from color import map_wavelength
-from solids import r7081
+from solids import r11708
 from geometry import Geometry
+import src
 
 nphotons = 1000
 
@@ -21,8 +22,14 @@ geometry.build(bits=8)
 cuda.init()
 
 try:
-    gputhread = GPUThread(5, geometry, jobs, output, 64)
-    gputhread.start()
+    #gputhread = GPUThread(5, geometry, jobs, output, 64)
+    #gputhread.start()
+
+    device = cuda.Device(5)
+    context = device.make_context()
+    module = SourceModule(src.kernel, options=['-I' + src.dir], no_extern_c=True, cache_dir=False)
+
+    propagate = module.get_function('propagate')
 
     job = create_job((0,0,0), nphotons)