summaryrefslogtreecommitdiff
path: root/xnt/tests/compilercollectiontests.py
diff options
context:
space:
mode:
Diffstat (limited to 'xnt/tests/compilercollectiontests.py')
-rw-r--r--xnt/tests/compilercollectiontests.py4
1 files changed, 3 insertions, 1 deletions
diff --git a/xnt/tests/compilercollectiontests.py b/xnt/tests/compilercollectiontests.py
index cb5b3fc..20d894f 100644
--- a/xnt/tests/compilercollectiontests.py
+++ b/xnt/tests/compilercollectiontests.py
@@ -97,7 +97,7 @@ class NvccTests(unittest.TestCase):
with open("temp/hello.cu", "w") as test_code:
test_code.write("""
__global__ void kernel(float *x) {
- int idx = threadIdx.x;
+ int idx = blockIdx.x;
x[idx] = 42;
}
int main() {
@@ -109,6 +109,8 @@ class NvccTests(unittest.TestCase):
kernel<<<128, 1>>>(dev_x);
cudaMemcpy(x, dev_x, size, cudaMemcpyDeviceToHost);
cudaFree(dev_x);
+ delete(x);
+ x = NULL;
}""")
def tearDown(self):
"""Test Case Teardown"""