Create a list of CudaGPU devices. Args: num (int): number of device to create. Returns: a list of swig converted CudaGPU devices.
(num)
| 66 | |
| 67 | |
| 68 | def create_cuda_gpus(num): |
| 69 | '''Create a list of CudaGPU devices. |
| 70 | |
| 71 | Args: |
| 72 | num (int): number of device to create. |
| 73 | Returns: |
| 74 | a list of swig converted CudaGPU devices. |
| 75 | ''' |
| 76 | assert singa.USE_CUDA, 'SINGA has not been compiled with CUDA enabled.' |
| 77 | return singa.Platform.CreateCudaGPUs(num) |
| 78 | |
| 79 | |
| 80 | def create_cuda_gpu(): |
nothing calls this directly
no test coverage detected