Workaround to support GTX 980

942d4d7a · Peter Eastman · f96c8313 · 942d4d7a
Commit 942d4d7a authored Oct 15, 2014 by Peter Eastman
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 0 deletions

platforms/cuda/src/CudaContext.cpp platforms/cuda/src/CudaContext.cpp +5 -0

No files found.
--- a/platforms/cuda/src/CudaContext.cpp
+++ b/platforms/cuda/src/CudaContext.cpp
@@ -136,6 +136,11 @@ CudaContext::CudaContext(const System& system, int deviceIndex, bool useBlocking
    this->deviceIndex = deviceIndex;
    int major, minor;
    CHECK_RESULT(cuDeviceComputeCapability(&major, &minor, device));
+    // This is a workaround to support GTX 980 with CUDA 6.5.  It reports its compute capability
+    // as 5.2, but the compiler doesn't support anything beyond 5.0.  We can remove this once
+    // CUDA 7.0 is released.
+    if (major == 5)
+        minor = 0;
    gpuArchitecture = intToString(major)+intToString(minor);
    computeCapability = major+0.1*minor;
    if ((useDoublePrecision || useMixedPrecision) && computeCapability < 1.3)