| 134 | #ifndef CPU_ONLY |
| 135 | inline static cublasHandle_t cublas_handle() { return Get().cublas_handle_; } |
| 136 | inline static curandGenerator_t curand_generator() { |
| 137 | return Get().curand_generator_; |
| 138 | } |
| 139 | #endif |
| 140 | |
| 141 | // Returns the mode: running on CPU or GPU. |