diff options
Diffstat (limited to 'test')
| -rw-r--r-- | test/linalg_test.py | 2 | ||||
| -rw-r--r-- | test/matrix_test.py | 3 | ||||
| -rw-r--r-- | test/rotate_test.py | 2 | ||||
| -rw-r--r-- | test/test_sample_cdf.py | 2 |
4 files changed, 5 insertions, 4 deletions
diff --git a/test/linalg_test.py b/test/linalg_test.py index 31688d9..23ebfa7 100644 --- a/test/linalg_test.py +++ b/test/linalg_test.py @@ -10,7 +10,7 @@ float3 = gpuarray.vec.float3 print 'device %s' % autoinit.device.name() current_directory = os.path.split(os.path.realpath(__file__))[0] -source_directory = current_directory + '/../src' +from chroma.cuda import srcdir as source_directory source = open(current_directory + '/linalg_test.cu').read() diff --git a/test/matrix_test.py b/test/matrix_test.py index c843025..3499945 100644 --- a/test/matrix_test.py +++ b/test/matrix_test.py @@ -10,7 +10,8 @@ float3 = gpuarray.vec.float3 print 'device %s' % autoinit.device.name() current_directory = os.path.split(os.path.realpath(__file__))[0] -source_directory = current_directory + '/../src' + +from chroma.cuda import srcdir as source_directory source = open(current_directory + '/matrix_test.cu').read() diff --git a/test/rotate_test.py b/test/rotate_test.py index 92eff84..7ac7804 100644 --- a/test/rotate_test.py +++ b/test/rotate_test.py @@ -18,7 +18,7 @@ def rotate(x, phi, n): print 'device %s' % autoinit.device.name() current_directory = os.path.split(os.path.realpath(__file__))[0] -source_directory = current_directory + '/../src' +from chroma.cuda import srcdir as source_directory source = open(current_directory + '/rotate_test.cu').read() diff --git a/test/test_sample_cdf.py b/test/test_sample_cdf.py index 2baa2ac..d12efa4 100644 --- a/test/test_sample_cdf.py +++ b/test/test_sample_cdf.py @@ -7,7 +7,7 @@ import ROOT import os current_directory = os.path.split(os.path.realpath(__file__))[0] -source_directory = current_directory + '/../src' +from chroma.cuda import srcdir as source_directory source = open(current_directory + '/test_sample_cdf.cu').read() |