summaryrefslogtreecommitdiff
path: root/test
diff options
context:
space:
mode:
authorStan Seibert <stan@mtrr.org>2011-09-16 19:21:57 -0400
committerStan Seibert <stan@mtrr.org>2011-09-16 19:21:57 -0400
commita2f7a3aed3eccb018dda989b5390371d6c0d6784 (patch)
tree0e05359ad9fde4aae38e730bd0d42b0483bf996b /test
parent9d792d1086f96fba91b9de6dddd73615c26652fd (diff)
downloadchroma-a2f7a3aed3eccb018dda989b5390371d6c0d6784.tar.gz
chroma-a2f7a3aed3eccb018dda989b5390371d6c0d6784.tar.bz2
chroma-a2f7a3aed3eccb018dda989b5390371d6c0d6784.zip
Add a srcdir attribute to the chroma.cuda module. This consolidates the calculation of the path to the CUDA source code in one place.
Diffstat (limited to 'test')
-rw-r--r--test/linalg_test.py2
-rw-r--r--test/matrix_test.py3
-rw-r--r--test/rotate_test.py2
-rw-r--r--test/test_sample_cdf.py2
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()