| 73 | } |
| 74 | |
| 75 | size_t getCudaAvailableMemory() |
| 76 | { |
| 77 | if ( CUDA_EXEC( cudaSetDevice( 0 ) ) != cudaSuccess ) |
| 78 | return 0; |
| 79 | size_t memFree = 0, memTot = 0; |
| 80 | if ( CUDA_EXEC( cudaMemGetInfo( &memFree, &memTot ) ) ) |
| 81 | return 0; |
| 82 | // minus extra 128 MB |
| 83 | return memFree - 128 * 1024 * 1024; |
| 84 | } |
| 85 | |
| 86 | size_t getCudaSafeMemoryLimit() |
| 87 | { |
no outgoing calls
no test coverage detected