Minor bug fix

8689a536 · Peter Eastman · 8d206abc · 8689a536
Commit 8689a536 authored Aug 26, 2009 by Peter Eastman
Show whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletion

platforms/cuda/src/kernels/cudaCompact.cu platforms/cuda/src/kernels/cudaCompact.cu +3 -1

No files found.
--- a/platforms/cuda/src/kernels/cudaCompact.cu
+++ b/platforms/cuda/src/kernels/cudaCompact.cu
@@ -184,8 +184,10 @@ __global__ void moveValidElementsScattered(const T* dgData,T* dgCompact,const un
 }
 void planCompaction(compactionPlan& d,bool stageOutput) {
+    int device;
+    cudaGetDevice(&device);
    cudaDeviceProp deviceProp;
-    cudaGetDeviceProperties(&deviceProp, 0);
+    cudaGetDeviceProperties(&deviceProp, device);
    d.nThreadBlocks = 16*deviceProp.multiProcessorCount;
    cudaMalloc((void**)&(d.dgBlockCounts), d.nThreadBlocks*sizeof(unsigned int));
    d.stageOutput = stageOutput;