Reapply changes from NVIDIA#404 that were lost in conflict resolution.

Fixes NVIDIA#510.
alliepiper · Jun 22, 2022 · 04a78a6 · 04a78a6
1 parent 92b501a
commit 04a78a6
Showing 1 changed file with 3 additions and 8 deletions.
diff --git a/cub/util_device.cuh b/cub/util_device.cuh
@@ -363,14 +363,9 @@ CUB_RUNTIME_FUNCTION inline cudaError_t PtxVersionUncached(int& ptx_version)
       (
         cudaFuncAttributes empty_kernel_attrs;
 
-        do
-        {
-          if (CubDebug(result = cudaFuncGetAttributes(&empty_kernel_attrs,
-                                                      empty_kernel)))
-          {
-            break;
-          }
-        } while (0);
+        result = cudaFuncGetAttributes(&empty_kernel_attrs,
+                                       reinterpret_cast<void*>(empty_kernel));
+        CubDebug(result);
 
         ptx_version = empty_kernel_attrs.ptxVersion * 10;
       ),