Added init_cpu() function to init CPU just to speedup the 1st detection (if GPU=0)

This commit is contained in:
AlexeyAB
2019-11-25 18:10:39 +03:00
parent dd4dcf20b4
commit 48914023da
5 changed files with 16 additions and 2 deletions

View File

@ -144,6 +144,8 @@ if hasGPU:
set_gpu = lib.cuda_set_device
set_gpu.argtypes = [c_int]
init_cpu = lib.init_cpu
make_image = lib.make_image
make_image.argtypes = [c_int, c_int, c_int]
make_image.restype = IMAGE