summaryrefslogtreecommitdiffstats
path: root/Tests/CudaOnly/ExportPTX
diff options
context:
space:
mode:
Diffstat (limited to 'Tests/CudaOnly/ExportPTX')
-rw-r--r--Tests/CudaOnly/ExportPTX/CMakeLists.txt1
-rw-r--r--Tests/CudaOnly/ExportPTX/kernelA.cu4
-rw-r--r--Tests/CudaOnly/ExportPTX/kernelB.cu3
3 files changed, 8 insertions, 0 deletions
diff --git a/Tests/CudaOnly/ExportPTX/CMakeLists.txt b/Tests/CudaOnly/ExportPTX/CMakeLists.txt
index f1667af..e97274d 100644
--- a/Tests/CudaOnly/ExportPTX/CMakeLists.txt
+++ b/Tests/CudaOnly/ExportPTX/CMakeLists.txt
@@ -11,6 +11,7 @@ list(SUBLIST CMAKE_CUDA_ARCHITECTURES 0 1 CMAKE_CUDA_ARCHITECTURES)
string(APPEND CMAKE_CUDA_ARCHITECTURES "-virtual")
add_library(CudaPTX OBJECT kernelA.cu kernelB.cu)
+target_compile_definitions(CudaPTX PRIVATE "CUDA_PTX_COMPILATION")
set_property(TARGET CudaPTX PROPERTY CUDA_PTX_COMPILATION ON)
#Test ObjectFiles with file(GENERATE)
diff --git a/Tests/CudaOnly/ExportPTX/kernelA.cu b/Tests/CudaOnly/ExportPTX/kernelA.cu
index fbe0d26..8967298 100644
--- a/Tests/CudaOnly/ExportPTX/kernelA.cu
+++ b/Tests/CudaOnly/ExportPTX/kernelA.cu
@@ -1,4 +1,8 @@
+#ifndef CUDA_PTX_COMPILATION
+# error "CUDA_PTX_COMPILATION define not provided"
+#endif
+
__global__ void kernelA(float* r, float* x, float* y, float* z, int size)
{
for (int i = threadIdx.x; i < size; i += blockDim.x) {
diff --git a/Tests/CudaOnly/ExportPTX/kernelB.cu b/Tests/CudaOnly/ExportPTX/kernelB.cu
index 11872e4..be4613a 100644
--- a/Tests/CudaOnly/ExportPTX/kernelB.cu
+++ b/Tests/CudaOnly/ExportPTX/kernelB.cu
@@ -1,4 +1,7 @@
+#ifndef CUDA_PTX_COMPILATION
+# error "CUDA_PTX_COMPILATION define not provided"
+#endif
__global__ void kernelB(float* r, float* x, float* y, float* z, int size)
{