def get_handle(): dev = cuda.get_device_id() if dev in _handles: return _handles[dev] handle = cudnn.create() _handles[dev] = handle return handle
def get_handle(): global _handles device = cuda.Device() handle = _handles.get(device.id, None) if handle is None: handle = cudnn.create() _handles[device.id] = handle return handle