Tensor Comprehensions
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Pages
tc::CudaGPUInfo Class Reference

#include <cuda_info.h>

Collaboration diagram for tc::CudaGPUInfo:

Public Member Functions

int NumberGPUs () const
 
int CurrentGPUId () const
 
void SynchronizeCurrentGPU () const
 
std::string GetGPUName (int id=-1) const
 
std::string GetCudaDeviceStr () const
 

Static Public Member Functions

static CudaGPUInfoGPUInfo ()
 

Public Attributes

std::vector< std::string > gpuNames_
 

Private Member Functions

 CudaGPUInfo (const std::vector< std::string > &gpuNames)
 

Constructor & Destructor Documentation

tc::CudaGPUInfo::CudaGPUInfo ( const std::vector< std::string > &  gpuNames)
inlineprivate

Member Function Documentation

int tc::CudaGPUInfo::CurrentGPUId ( ) const
std::string tc::CudaGPUInfo::GetCudaDeviceStr ( ) const
std::string tc::CudaGPUInfo::GetGPUName ( int  id = -1) const
static CudaGPUInfo& tc::CudaGPUInfo::GPUInfo ( )
static
int tc::CudaGPUInfo::NumberGPUs ( ) const
void tc::CudaGPUInfo::SynchronizeCurrentGPU ( ) const

Member Data Documentation

std::vector<std::string> tc::CudaGPUInfo::gpuNames_

The documentation for this class was generated from the following file: