Searched refs:CudaVersion (Results 1 – 6 of 6) sorted by relevance
9 const char *CudaVersionToString(CudaVersion V) { in CudaVersionToString()11 case CudaVersion::UNKNOWN: in CudaVersionToString()13 case CudaVersion::CUDA_70: in CudaVersionToString()15 case CudaVersion::CUDA_75: in CudaVersionToString()17 case CudaVersion::CUDA_80: in CudaVersionToString()149 CudaVersion MinVersionForCudaArch(CudaArch A) { in MinVersionForCudaArch()152 return CudaVersion::UNKNOWN; in MinVersionForCudaArch()162 return CudaVersion::CUDA_70; in MinVersionForCudaArch()166 return CudaVersion::CUDA_80; in MinVersionForCudaArch()
19 enum class CudaVersion { enum25 const char *CudaVersionToString(CudaVersion V);73 CudaVersion MinVersionForCudaArch(CudaArch A);
1406 struct CudaVersion { struct1408 explicit CudaVersion(const std::string& version_name) { in CudaVersion() function1419 CudaVersion() {} in CudaVersion() argument1420 bool operator<(const CudaVersion& other) const { in operator <() argument1427 const CudaVersion& version) { in operator <<()1435 std::vector<CudaVersion> supported_cuda_compute_capabilities = {1438 std::vector<CudaVersion> GetSupportedCudaComputeCapabilities() { in GetSupportedCudaComputeCapabilities()1452 cuda_caps.push_back(CudaVersion(capability)); in GetSupportedCudaComputeCapabilities()1586 CudaVersion min_supported_capability = *std::min_element( in GetValidDeviceIds()1616 CudaVersion device_capability; in GetValidDeviceIds()
167 CudaVersion Version = CudaVersion::UNKNOWN;195 CudaVersion version() const { return Version; } in version()
1728 static CudaVersion ParseCudaVersionFile(llvm::StringRef V) { in ParseCudaVersionFile()1730 return CudaVersion::UNKNOWN; in ParseCudaVersionFile()1737 return CudaVersion::UNKNOWN; in ParseCudaVersionFile()1742 return CudaVersion::CUDA_70; in ParseCudaVersionFile()1745 return CudaVersion::CUDA_75; in ParseCudaVersionFile()1747 return CudaVersion::CUDA_80; in ParseCudaVersionFile()1748 return CudaVersion::UNKNOWN; in ParseCudaVersionFile()1820 Version = CudaVersion::CUDA_70; in init()1832 if (Arch == CudaArch::UNKNOWN || Version == CudaVersion::UNKNOWN || in CheckCudaVersionSupportsArch()
512 set(TF_CUDA_CAP "${TF_CUDA_CAP},CudaVersion(\"${NVCC_ARCH_MAJOR}.${NVCC_ARCH_MINOR}\")")514 set(TF_CUDA_CAP "CudaVersion(\"${NVCC_ARCH_MAJOR}.${NVCC_ARCH_MINOR}\")")