Skip to content
Snippets Groups Projects
Commit 48040a03 authored by Eric Wait's avatar Eric Wait
Browse files

Renamed a variable in device count

parent d6a03fb7
No related tags found
No related merge requests found
......@@ -3,12 +3,12 @@
int cMemoryStats(size_t** stats)
{
int cnt = 0;
cudaGetDeviceCount(&cnt);
int deviceCount = 0;
cudaGetDeviceCount(&deviceCount);
*stats = new size_t[cnt*2];
*stats = new size_t[deviceCount*2];
for(int i=0; i<cnt; ++i)
for(int i=0; i<deviceCount; ++i)
{
cudaSetDevice(i);
......@@ -18,5 +18,5 @@ int cMemoryStats(size_t** stats)
(*stats)[i*2+1] = availMem;
}
return cnt;
}
\ No newline at end of file
return deviceCount;
}
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment