Function nnHandle

src/backend/cuda/platform.cpp:525–539

#ifdef WITH_CUDNN
cudnnHandle_t nnHandle() {
    // Keep the getCudnnPlugin call here because module loading can throw an
    // exception the first time its called. We want to avoid that because
    // the unique handle object is marked noexcept and could terminate. if
    // the module is not loaded correctly
    static cudnnModule keep_me_to_avoid_exceptions_exceptions =
        getCudnnPlugin();
    static unique_handle<cudnnHandle_t> *handle =
        nnManager(getActiveDeviceId());
    if (*handle) {
        return *handle;
    } else {
        AF_ERROR("Error Initializing cuDNN\n", AF_ERR_RUNTIME);
    }
}
#endif

SolveHandle solverDnHandle() { return *cusolverManager(getActiveDeviceId()); }