Create a list of CudaGPU devices. Args: device_ids (list): a list of GPU card IDs. Returns: a list of swig converted CudaGPU devices.
(device_ids)
| 88 | |
| 89 | |
| 90 | def create_cuda_gpus_on(device_ids): |
| 91 | '''Create a list of CudaGPU devices. |
| 92 | |
| 93 | Args: |
| 94 | device_ids (list): a list of GPU card IDs. |
| 95 | |
| 96 | Returns: |
| 97 | a list of swig converted CudaGPU devices. |
| 98 | ''' |
| 99 | assert singa.USE_CUDA, 'SINGA has not been compiled with CUDA enabled.' |
| 100 | return singa.Platform.CreateCudaGPUsOn(device_ids) |
| 101 | |
| 102 | |
| 103 | def create_cuda_gpu_on(device_id): |
no test coverage detected