diff options
Diffstat (limited to 'tests/linalg_test.py')
| -rw-r--r-- | tests/linalg_test.py | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/tests/linalg_test.py b/tests/linalg_test.py index 44c4b52..31688d9 100644 --- a/tests/linalg_test.py +++ b/tests/linalg_test.py @@ -1,3 +1,4 @@ +import os import numpy as np from pycuda import autoinit from pycuda.compiler import SourceModule @@ -8,9 +9,12 @@ float3 = gpuarray.vec.float3 print 'device %s' % autoinit.device.name() -source = open('../linalg.h').read() + open('linalg_test.cu').read() +current_directory = os.path.split(os.path.realpath(__file__))[0] +source_directory = current_directory + '/../src' -mod = SourceModule(source, no_extern_c=True, arch='sm_13') +source = open(current_directory + '/linalg_test.cu').read() + +mod = SourceModule(source, options=['-I' + source_directory], no_extern_c=True, cache_dir=False) float3add = mod.get_function('float3add') float3addequal = mod.get_function('float3addequal') |