Update OpenCLContext.cpp (#3917)

a1a32466 · Philip Turner · GitHub · 6b51520c · a1a32466
Unverified Commit a1a32466 authored Jan 31, 2023 by Philip Turner Committed by GitHub Jan 31, 2023
Show whitespace changes
Inline Side-by-side

Showing with 5 additions and 1 deletion

platforms/opencl/src/OpenCLContext.cpp platforms/opencl/src/OpenCLContext.cpp +5 -1

No files found.
--- a/platforms/opencl/src/OpenCLContext.cpp
+++ b/platforms/opencl/src/OpenCLContext.cpp
@@ -211,7 +211,11 @@ OpenCLContext::OpenCLContext(const System& system, int platformIndex, int device
            throw OpenMMException("This device does not support double precision");
        string vendor = device.getInfo<CL_DEVICE_VENDOR>();
        int numThreadBlocksPerComputeUnit = 6;
-        if (vendor.size() >= 6 && vendor.substr(0, 6) == "NVIDIA") {
+      
+        if (vendor.size() >= 5 && vendor.substr(0, 5) == "Apple") {
+            simdWidth = 32;
+        }
+        else if (vendor.size() >= 6 && vendor.substr(0, 6) == "NVIDIA") {
            compilationDefines["WARPS_ARE_ATOMIC"] = "";
            simdWidth = 32;
            if (device.getInfo<CL_DEVICE_EXTENSIONS>().find("cl_nv_device_attribute_query") != string::npos) {