diff options
Diffstat (limited to 'nerv/lib/matrix/cumatrix.h')
-rw-r--r-- | nerv/lib/matrix/cumatrix.h | 14 |
1 files changed, 14 insertions, 0 deletions
diff --git a/nerv/lib/matrix/cumatrix.h b/nerv/lib/matrix/cumatrix.h index e6def66..e53c702 100644 --- a/nerv/lib/matrix/cumatrix.h +++ b/nerv/lib/matrix/cumatrix.h @@ -4,4 +4,18 @@ void nerv_cumatrix_print_profile(); void nerv_cumatrix_clear_profile(); void nerv_cumatrix_init(); + +void nerv_set_cublas_handle(); + +typedef struct CuContext +{ + cublasHandle_t cublas_handle; + cudaEvent_t profile_start, profile_stop; + HashMap *profile; + pthread_t pid; + int refcount; +}CuContext; + +extern const char *nerv_context_tname; + #endif |