cory-master/BabelStream
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Use device info to select CUDA device

Browse Source
This commit is contained in:
Tom Deakin 2016-05-03 11:15:38 +01:00
parent 3462e61c16
commit fd121c2467
2 changed files with 16 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
src/CUDAStream.cu
View File

@ -12,8 +12,22 @@ void check_error(void)
}
template <class T>
CUDAStream<T>::CUDAStream(const unsigned int ARRAY_SIZE)
CUDAStream<T>::CUDAStream(const unsigned int ARRAY_SIZE, const int device_index)
{
// Set device
int count;
cudaGetDeviceCount(&count);
check_error();
if (device_index >= count)
throw std::runtime_error("Invalid device index");
cudaSetDevice(device_index);
check_error();
// Print out device information
std::cout << "Using OpenCL device " << getDeviceName(device_index) << std::endl;
std::cout << "Driver: " << getDeviceDriver(device_index) << std::endl;
array_size = ARRAY_SIZE;
// Check buffers fit on the device

2
src/CUDAStream.h
View File

@ -20,7 +20,7 @@ class CUDAStream : public Stream<T>
public:
CUDAStream(const unsigned int);
CUDAStream(const unsigned int, const int);
~CUDAStream();
virtual void copy() override;