def get_pytest_marks(self):
marks = []
if self.use_chainerx:
marks.append(attr.chainerx)
backend_name, device_index = self.chainerx_device.split(':')
device_index = int(device_index)
if backend_name == 'cuda':
marks.append(attr.gpu)
if device_index >= 1:
marks.append(attr.multi_gpu(device_index + 1))
elif self.use_cuda:
marks.append(attr.gpu)
if self.use_cudnn != 'never':
marks.append(attr.cudnn)
if self.cuda_device >= 1:
marks.append(attr.multi_gpu(self.cuda_device + 1))
else:
if self.use_ideep != 'never':
marks.append(attr.ideep)
assert all(callable(_) for _ in marks)
return marks
def get_pytest_marks(self):
marks = []
if self.use_chainerx:
marks.append(attr.chainerx)
backend_name, device_index = self.chainerx_device.split(':')
device_index = int(device_index)
if backend_name == 'cuda':
marks.append(attr.gpu)
if device_index >= 1:
marks.append(attr.multi_gpu(device_index + 1))
elif self.use_cuda:
marks.append(attr.gpu)
if self.use_cudnn != 'never':
marks.append(attr.cudnn)
if self.cuda_device >= 1:
marks.append(attr.multi_gpu(self.cuda_device + 1))
else:
if self.use_ideep != 'never':
marks.append(attr.ideep)
assert all(callable(_) for _ in marks)
return marks