コード例 #1
0
ファイル: GpuContext.py プロジェクト: Sandy4321/quagga 
 def __del__(self):
     with cudart.device(self.device_id):
         cudart.cuda_stream_destroy(self.cuda_stream)
コード例 #2
0
ファイル: GpuContext.py プロジェクト: Sandy4321/quagga 
 def __init__(self, device_id=None):
     with cudart.device(device_id):
         self.device_id = cudart.cuda_get_device()
         self.cuda_stream = cudart.ct_cuda_stream()
         cudart.cuda_stream_create(self.cuda_stream)
コード例 #3
0
ファイル: GpuContext.py プロジェクト: Sandy4321/quagga 
def _create_cudnn_handle(device_id):
    with cudart.device(device_id):
        handle = cudnn.ct_cudnn_handle()
        cudnn.create(handle)
    return handle
コード例 #4
0
ファイル: GpuContext.py プロジェクト: datatalking/quagga 
 def __del__(self):
     with cudart.device(self.device_id):
         cudart.cuda_stream_destroy(self.cuda_stream)
コード例 #5
0
ファイル: GpuContext.py プロジェクト: datatalking/quagga 
 def __init__(self, device_id=None):
     with cudart.device(device_id):
         self.device_id = cudart.cuda_get_device()
         self.cuda_stream = cudart.ct_cuda_stream()
         cudart.cuda_stream_create(self.cuda_stream)
コード例 #6
0
ファイル: GpuContext.py プロジェクト: datatalking/quagga 
def _create_cudnn_handle(device_id):
    with cudart.device(device_id):
        handle = cudnn.ct_cudnn_handle()
        cudnn.create(handle)
    return handle