diff options
Diffstat (limited to 'geometry.py')
| -rw-r--r-- | geometry.py | 16 |
1 files changed, 11 insertions, 5 deletions
diff --git a/geometry.py b/geometry.py index d13fc9f..3181176 100644 --- a/geometry.py +++ b/geometry.py @@ -404,25 +404,31 @@ class Geometry(object): self.node_map_gpu = cuda.to_device(self.node_map) self.node_length_gpu = cuda.to_device(self.node_length) + print 'Device usage:' + print 'vertices:', vertices.nbytes + print 'triangles:', triangles.nbytes + print 'lower_bounds:', lower_bounds.nbytes + print 'upper_bounds:', upper_bounds.nbytes + print 'node_map:', self.node_map.nbytes + print 'node_length:', self.node_length.nbytes + set_pointer = module.get_function('set_pointer') - set_pointer(self.triangles_gpu, block=(1,1,1), grid=(1,1)) + set_pointer(self.triangles_gpu, self.vertices_gpu, + block=(1,1,1), grid=(1,1)) - vertices_tex = module.get_texref('vertices') lower_bounds_tex = module.get_texref('lower_bounds') upper_bounds_tex = module.get_texref('upper_bounds') node_map_tex = module.get_texref('node_map') node_length_tex = module.get_texref('node_length') - vertices_tex.set_address(self.vertices_gpu, vertices.nbytes) lower_bounds_tex.set_address(self.lower_bounds_gpu, lower_bounds.nbytes) upper_bounds_tex.set_address(self.upper_bounds_gpu, upper_bounds.nbytes) node_map_tex.set_address(self.node_map_gpu, self.node_map.nbytes) node_length_tex.set_address(self.node_length_gpu, self.node_length.nbytes) - vertices_tex.set_format(cuda.array_format.FLOAT, 4) lower_bounds_tex.set_format(cuda.array_format.FLOAT, 4) upper_bounds_tex.set_format(cuda.array_format.FLOAT, 4) node_map_tex.set_format(cuda.array_format.UNSIGNED_INT32, 1) node_length_tex.set_format(cuda.array_format.UNSIGNED_INT32, 1) - return [vertices_tex, lower_bounds_tex, upper_bounds_tex, node_map_tex, node_length_tex] + return [lower_bounds_tex, upper_bounds_tex, node_map_tex, node_length_tex] |