Check if platform index is out of range

include/util.hpp

@@ -67,3 +67,20 @@ static bool checkDeviceForExtension(
 
     return supported;
 }
+
+static bool checkPlatformIndex(
+    const std::vector<cl::Platform>& platforms,
+    int platformIndex)
+{
+    if (platforms.size() == 0) {
+        fprintf(stderr, "Error: No OpenCL platforms found.\n");
+        return false;
+    }
+    if (platformIndex >= (int)platforms.size()) {
+        fprintf(stderr, "Error: Invalid platform index %d specified (max %d)\n",
+            platformIndex,
+            (int)(platforms.size() - 1) );
+        return false;
+    }
+    return true;
+}

samples/01_copybuffer/main.cpp

@@ -8,6 +8,8 @@
 
 #include <CL/opencl.hpp>
 
+#include "util.hpp"
+
 const size_t    gwx = 1024*1024;
 
 int main(
@@ -41,6 +43,10 @@ int main(
     std::vector<cl::Platform> platforms;
     cl::Platform::get(&platforms);
 
+    if (!checkPlatformIndex(platforms, platformIndex)) {
+        return -1;
+    }
+
     printf("Running on platform: %s\n",
         platforms[platformIndex].getInfo<CL_PLATFORM_NAME>().c_str() );
 

samples/02_copybufferkernel/main.cpp

@@ -8,6 +8,8 @@
 
 #include <CL/opencl.hpp>
 
+#include "util.hpp"
+
 const size_t    gwx = 1024*1024;
 
 static const char kernelString[] = R"CLC(
@@ -49,6 +51,10 @@ int main(
     std::vector<cl::Platform> platforms;
     cl::Platform::get(&platforms);
 
+    if (!checkPlatformIndex(platforms, platformIndex)) {
+        return -1;
+    }
+
     printf("Running on platform: %s\n",
         platforms[platformIndex].getInfo<CL_PLATFORM_NAME>().c_str() );
 

samples/03_mandelbrot/main.cpp

@@ -11,6 +11,8 @@
 
 #include <CL/opencl.hpp>
 
+#include "util.hpp"
+
 const char* filename = "mandelbrot.bmp";
 
 const cl_uint width = 768;
@@ -83,6 +85,10 @@ int main(
     std::vector<cl::Platform> platforms;
     cl::Platform::get(&platforms);
 
+    if (!checkPlatformIndex(platforms, platformIndex)) {
+        return -1;
+    }
+
     printf("Running on platform: %s\n",
         platforms[platformIndex].getInfo<CL_PLATFORM_NAME>().c_str() );
 

samples/04_julia/main.cpp

@@ -13,6 +13,8 @@
 
 #include <chrono>
 
+#include "util.hpp"
+
 const char* filename = "julia.bmp";
 
 const float cr = -0.123f;
@@ -104,6 +106,10 @@ int main(
     std::vector<cl::Platform> platforms;
     cl::Platform::get(&platforms);
 
+    if (!checkPlatformIndex(platforms, platformIndex)) {
+        return -1;
+    }
+
     printf("Running on platform: %s\n",
         platforms[platformIndex].getInfo<CL_PLATFORM_NAME>().c_str() );
 

samples/04_sobel/main.cpp

@@ -13,6 +13,8 @@
 
 #include <chrono>
 
+#include "util.hpp"
+
 const char* filename = "sobel.bmp";
 
 const float cr = -0.123f;
@@ -151,6 +153,10 @@ int main(
     std::vector<cl::Platform> platforms;
     cl::Platform::get(&platforms);
 
+    if (!checkPlatformIndex(platforms, platformIndex)) {
+        return -1;
+    }
+
     printf("Running on platform: %s\n",
         platforms[platformIndex].getInfo<CL_PLATFORM_NAME>().c_str() );
 

samples/05_kernelfromfile/main.cpp

@@ -11,6 +11,8 @@
 #include <fstream>
 #include <string>
 
+#include "util.hpp"
+
 static std::string readStringFromFile(
     const std::string& filename )
 {
@@ -71,6 +73,10 @@ int main(
     std::vector<cl::Platform> platforms;
     cl::Platform::get(&platforms);
 
+    if (!checkPlatformIndex(platforms, platformIndex)) {
+        return -1;
+    }
+
     printf("Running on platform: %s\n",
         platforms[platformIndex].getInfo<CL_PLATFORM_NAME>().c_str() );
 

samples/05_spirvkernelfromfile/main.cpp

@@ -134,6 +134,10 @@ int main(
     std::vector<cl::Platform> platforms;
     cl::Platform::get(&platforms);
 
+    if (!checkPlatformIndex(platforms, platformIndex)) {
+        return -1;
+    }
+
     printf("Running on platform: %s\n",
         platforms[platformIndex].getInfo<CL_PLATFORM_NAME>().c_str() );
 

samples/06_ndrangekernelfromfile/main.cpp

@@ -11,6 +11,8 @@
 #include <fstream>
 #include <string>
 
+#include "util.hpp"
+
 static std::string readStringFromFile(
     const std::string& filename )
 {
@@ -86,6 +88,10 @@ int main(
     std::vector<cl::Platform> platforms;
     cl::Platform::get(&platforms);
 
+    if (!checkPlatformIndex(platforms, platformIndex)) {
+        return -1;
+    }
+
     printf("Running on platform: %s\n",
         platforms[platformIndex].getInfo<CL_PLATFORM_NAME>().c_str() );
 

samples/10_queueexperiments/main.cpp

@@ -413,6 +413,10 @@ int main(
     std::vector<cl::Platform> platforms;
     cl::Platform::get(&platforms);
 
+    if (!checkPlatformIndex(platforms, platformIndex)) {
+        return -1;
+    }
+
     printf("Running on platform: %s\n",
         platforms[platformIndex].getInfo<CL_PLATFORM_NAME>().c_str() );
 

samples/11_semaphores/main.cpp

@@ -71,6 +71,10 @@ int main(
     std::vector<cl::Platform> platforms;
     cl::Platform::get(&platforms);
 
+    if (!checkPlatformIndex(platforms, platformIndex)) {
+        return -1;
+    }
+
     printf("Running on platform: %s\n",
         platforms[platformIndex].getInfo<CL_PLATFORM_NAME>().c_str() );
 

samples/12_commandbuffers/main.cpp

@@ -97,6 +97,10 @@ int main(
     std::vector<cl::Platform> platforms;
     cl::Platform::get(&platforms);
 
+    if (!checkPlatformIndex(platforms, platformIndex)) {
+        return -1;
+    }
+
     printf("Running on platform: %s\n",
         platforms[platformIndex].getInfo<CL_PLATFORM_NAME>().c_str() );
 

samples/12_commandbufferspp/main.cpp

@@ -92,6 +92,10 @@ int main(
     std::vector<cl::Platform> platforms;
     cl::Platform::get(&platforms);
 
+    if (!checkPlatformIndex(platforms, platformIndex)) {
+        return -1;
+    }
+
     printf("Running on platform: %s\n",
         platforms[platformIndex].getInfo<CL_PLATFORM_NAME>().c_str() );
 

samples/13_mutablecommandbuffers/main.cpp

@@ -114,6 +114,10 @@ int main(
     std::vector<cl::Platform> platforms;
     cl::Platform::get(&platforms);
 
+    if (!checkPlatformIndex(platforms, platformIndex)) {
+        return -1;
+    }
+
     printf("Running on platform: %s\n",
         platforms[platformIndex].getInfo<CL_PLATFORM_NAME>().c_str() );
 

samples/14_ooqcommandbuffers/main.cpp

@@ -67,6 +67,10 @@ int main(
     std::vector<cl::Platform> platforms;
     cl::Platform::get(&platforms);
 
+    if (!checkPlatformIndex(platforms, platformIndex)) {
+        return -1;
+    }
+
     printf("Running on platform: %s\n",
         platforms[platformIndex].getInfo<CL_PLATFORM_NAME>().c_str() );
 

samples/15_mutablecommandbufferasserts/main.cpp

@@ -67,6 +67,10 @@ int main(
     std::vector<cl::Platform> platforms;
     cl::Platform::get(&platforms);
 
+    if (!checkPlatformIndex(platforms, platformIndex)) {
+        return -1;
+    }
+
     printf("Running on platform: %s\n",
         platforms[platformIndex].getInfo<CL_PLATFORM_NAME>().c_str() );
 

samples/16_floatatomics/main.cpp

@@ -124,6 +124,10 @@ int main(
     std::vector<cl::Platform> platforms;
     cl::Platform::get(&platforms);
 
+    if (!checkPlatformIndex(platforms, platformIndex)) {
+        return -1;
+    }
+
     printf("Running on platform: %s\n",
         platforms[platformIndex].getInfo<CL_PLATFORM_NAME>().c_str() );
 

samples/images/00_enumimageformats/main.cpp

@@ -10,6 +10,8 @@
 
 #include <algorithm>
 
+#include "util.hpp"
+
 #define CASE_TO_STRING(_e) case _e: return #_e;
 
 const char* mem_object_type_to_string(cl_mem_object_type mem_object_type)
@@ -159,6 +161,10 @@ int main(
     std::vector<cl::Platform> platforms;
     cl::Platform::get(&platforms);
 
+    if (!checkPlatformIndex(platforms, platformIndex)) {
+        return -1;
+    }
+
     printf("Querying platform: %s\n",
         platforms[platformIndex].getInfo<CL_PLATFORM_NAME>().c_str() );
 

samples/opengl/00_juliagl/main.cpp

@@ -493,6 +493,10 @@ int main(
     std::vector<cl::Platform> platforms;
     cl::Platform::get(&platforms);
 
+    if (!checkPlatformIndex(platforms, platformIndex)) {
+        return -1;
+    }
+
     printf("Running on platform: %s\n",
         platforms[platformIndex].getInfo<CL_PLATFORM_NAME>().c_str() );
 

samples/opengl/01_nbodygl/main.cpp

@@ -280,6 +280,10 @@ int main(
     std::vector<cl::Platform> platforms;
     cl::Platform::get(&platforms);
 
+    if (!checkPlatformIndex(platforms, platformIndex)) {
+        return -1;
+    }
+
     printf("Running on platform: %s\n",
         platforms[platformIndex].getInfo<CL_PLATFORM_NAME>().c_str() );
 

samples/opengl/02_sobelgl/main.cpp

@@ -540,6 +540,10 @@ int main(
     std::vector<cl::Platform> platforms;
     cl::Platform::get(&platforms);
 
+    if (!checkPlatformIndex(platforms, platformIndex)) {
+        return -1;
+    }
+
     printf("Running on platform: %s\n",
         platforms[platformIndex].getInfo<CL_PLATFORM_NAME>().c_str() );
 

samples/python/01_copybuffer/copybuffer.py

@@ -7,6 +7,7 @@
 import numpy as np
 import pyopencl as cl
 import argparse
+import sys
 
 gwx = 1024 * 1024
 
@@ -20,6 +21,8 @@
     deviceIndex = args.device
 
     platforms = cl.get_platforms()
+    if platformIndex >= len(platforms):
+        sys.exit('Invalid platform index: {}'.format(platformIndex))
     print('Running on platform: ' + platforms[platformIndex].get_info(cl.platform_info.NAME))
 
     devices = platforms[platformIndex].get_devices()

samples/python/02_copybufferkernel/copybufferkernel.py

@@ -7,6 +7,7 @@
 import numpy as np
 import pyopencl as cl
 import argparse
+import sys
 
 gwx = 1024 * 1024
 
@@ -28,6 +29,8 @@
     deviceIndex = args.device
 
     platforms = cl.get_platforms()
+    if platformIndex >= len(platforms):
+        sys.exit('Invalid platform index: {}'.format(platformIndex))
     print('Running on platform: ' + platforms[platformIndex].get_info(cl.platform_info.NAME))
 
     devices = platforms[platformIndex].get_devices()

samples/python/03_mandelbrot/mandelbrot.py

@@ -10,6 +10,7 @@
 import pyopencl as cl
 import argparse
 import PIL
+import sys
 
 filename = 'mandelbrot.bmp'
 
@@ -63,6 +64,8 @@
     deviceIndex = args.device
 
     platforms = cl.get_platforms()
+    if platformIndex >= len(platforms):
+        sys.exit('Invalid platform index: {}'.format(platformIndex))
     print('Running on platform: ' + platforms[platformIndex].get_info(cl.platform_info.NAME))
 
     devices = platforms[platformIndex].get_devices()

samples/python/04_julia/julia.py

@@ -10,6 +10,7 @@
 import pyopencl as cl
 import argparse
 import PIL
+import sys
 import time
 
 filename = 'julia.bmp'
@@ -87,6 +88,8 @@
     lwy = args.lwy
 
     platforms = cl.get_platforms()
+    if platformIndex >= len(platforms):
+        sys.exit('Invalid platform index: {}'.format(platformIndex))
     print('Running on platform: ' + platforms[platformIndex].get_info(cl.platform_info.NAME))
 
     devices = platforms[platformIndex].get_devices()