diff options
Diffstat (limited to 'gpu.py')
-rw-r--r-- | gpu.py | 6 |
1 files changed, 3 insertions, 3 deletions
@@ -240,9 +240,9 @@ class GPU(object): assert len(photons.polarizations) == self.nphotons assert len(photons.wavelengths) == self.nphotons - self.positions_gpu = gpuarray.to_gpu(to_float3(photons.positions))#.astype(np.float32).view(gpuarray.vec.float3)) - self.directions_gpu = gpuarray.to_gpu(to_float3(photons.directions))#.astype(np.float32).view(gpuarray.vec.float3)) - self.polarizations_gpu = gpuarray.to_gpu(to_float3(photons.polarizations))#.astype(np.float32).view(gpuarray.vec.float3)) + self.positions_gpu = gpuarray.to_gpu(to_float3(photons.positions)) + self.directions_gpu = gpuarray.to_gpu(to_float3(photons.directions)) + self.polarizations_gpu = gpuarray.to_gpu(to_float3(photons.polarizations)) self.wavelengths_gpu = gpuarray.to_gpu(photons.wavelengths.astype(np.float32)) if photons.times is not None: |