ext/cumo/cumo.c in cumo-0.2.5 vs ext/cumo/cumo.c in cumo-0.3.0.pre1
- old
+ new
@@ -31,10 +31,12 @@
void Init_cumo_na_struct();
void Init_cumo_cuda_driver();
void Init_cumo_cuda_memory_pool();
void Init_cumo_cuda_runtime();
void Init_cumo_cuda_nvrtc();
+void Init_cumo_cuda_cublas();
+void Init_cumo_cuda_cudnn();
void
cumo_debug_breakpoint(void)
{
/* */
@@ -165,6 +167,9 @@
Init_cumo_cuda_driver();
Init_cumo_cuda_memory_pool();
Init_cumo_cuda_runtime();
Init_cumo_cuda_nvrtc();
+
+ Init_cumo_cuda_cublas();
+ Init_cumo_cuda_cudnn();
}