diff options
| author | Willem Jan Palenstijn <wjp@usecode.org> | 2017-10-17 17:20:40 +0200 |
|---|---|---|
| committer | Willem Jan Palenstijn <wjp@usecode.org> | 2017-10-17 17:20:40 +0200 |
| commit | 17ebcfefa7c42cff61a81bf12a915983b330ed15 (patch) | |
| tree | d0bf19b4e049e97d5de6c666f31fd8bf621105d3 /cuda/2d/util.h | |
| parent | b3a80d71aa9a80223729b8cb47a55297f985faaf (diff) | |
| download | astra-17ebcfefa7c42cff61a81bf12a915983b330ed15.tar.gz astra-17ebcfefa7c42cff61a81bf12a915983b330ed15.tar.bz2 astra-17ebcfefa7c42cff61a81bf12a915983b330ed15.tar.xz astra-17ebcfefa7c42cff61a81bf12a915983b330ed15.zip | |
Add astra.astra.get_gpu_info utility function
Diffstat (limited to 'cuda/2d/util.h')
| -rw-r--r-- | cuda/2d/util.h | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/cuda/2d/util.h b/cuda/2d/util.h index 875aae3..31fcfbd 100644 --- a/cuda/2d/util.h +++ b/cuda/2d/util.h @@ -30,6 +30,7 @@ along with the ASTRA Toolbox. If not, see <http://www.gnu.org/licenses/>. #include <cuda.h> #include <driver_types.h> +#include <string> #ifdef _MSC_VER @@ -92,6 +93,9 @@ void reportCudaError(cudaError_t err); float dotProduct2D(float* D_data, unsigned int pitch, unsigned int width, unsigned int height); +// Return string with CUDA device number, name and memory size. +// Use device == -1 to get info for the current device. +_AstraExport std::string getCudaDeviceString(int device); } |