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