Use std::iota in CUDATestDriver::target_devices

NVIDIA · Aug 13, 2020 · ed6b727 · ed6b727
1 parent 441a98f
commit ed6b727
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/testing/unittest/cuda/testframework.cu b/testing/unittest/cuda/testframework.cu
@@ -2,6 +2,7 @@
 #include <unittest/cuda/testframework.h>
 #include <thrust/system/cuda/memory.h>
 #include <cuda_runtime.h>
+#include <numeric>
 
 __global__ void dummy_kernel() {}
 
@@ -80,9 +81,7 @@ std::vector<int> CUDATestDriver::target_devices(const ArgumentMap &kwargs)
     cudaGetDeviceCount(&count);
 
     result.resize(count);
-    // XXX iota is not available in c++03
-    for(int i = 0; i < count; ++i)
-      result[i] = i;
+    std::iota(result.begin(), result.end(), 0);
   }
   else
   {