diff --git a/pystencils_tests/test_cudagpu.py b/pystencils_tests/test_cudagpu.py index 77790229f48b736258d343a4670c06e66a417838..f5efd517f0457ffc71d5c8015dd5d4f92aef47ea 100644 --- a/pystencils_tests/test_cudagpu.py +++ b/pystencils_tests/test_cudagpu.py @@ -1,4 +1,5 @@ import numpy as np +import pycuda.autoinit import pycuda.gpuarray as gpuarray import sympy as sp from scipy.ndimage import convolve