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

Implement some CUDA routines for device info

Browse Source
This commit is contained in:
Tom Deakin 2016-04-28 23:06:06 +01:00
parent a1cab96c57
commit d1f8cd1b48
3 changed files with 30 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
src/CUDAStream.cu
View File

@ -134,6 +134,28 @@ void CUDAStream<T>::triad()
check_error(); check_error();
} }
template <class T>
std::string CUDAStream<T>::getDeviceName(const int device)
{
cudaSetDevice(device);
check_error();
cudaDeviceProp props;
cudaGetDeviceProperties(&props, device);
check_error();
return std::string(props.name);
}
template <class T>
std::string CUDAStream<T>::getDeviceDriver(const int device)
{
cudaSetDevice(device);
check_error();
int driver;
cudaDriverGetVersion(&driver);
check_error();
return std::to_string(driver);
}
template class CUDAStream<float>; template class CUDAStream<float>;
template class CUDAStream<double>; template class CUDAStream<double>;

4
src/CUDAStream.h
View File

@ -31,5 +31,9 @@ class CUDAStream : public Stream<T>
virtual void write_arrays(const std::vector<T>& a, const std::vector<T>& b, const std::vector<T>& c) override; virtual void write_arrays(const std::vector<T>& a, const std::vector<T>& b, const std::vector<T>& c) override;
virtual void read_arrays(std::vector<T>& a, std::vector<T>& b, std::vector<T>& c) override; virtual void read_arrays(std::vector<T>& a, std::vector<T>& b, std::vector<T>& c) override;
static void listDevices(void);
static std::string getDeviceName(const int device);
static std::string getDeviceDriver(const int device);
}; };

7
src/Stream.h
View File

@ -2,6 +2,7 @@
#pragma once #pragma once
#include <vector> #include <vector>
#include <string>
template <class T> template <class T>
class Stream class Stream
@ -19,9 +20,9 @@ class Stream
virtual void read_arrays(std::vector<T>& a, std::vector<T>& b, std::vector<T>& c) = 0; virtual void read_arrays(std::vector<T>& a, std::vector<T>& b, std::vector<T>& c) = 0;
// Implementation specific device functions // Implementation specific device functions
static std::vector<int> getDeviceList(); static void listDevices(void);
static std::vector<int> getDeviceName(); static std::string getDeviceName(const int);
static std::vector<int> getDeviceDriver(); static std::string getDeviceDriver(const int);
}; };