tests/performance/api/device.py


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15


from . import TestEnvironment

class TestDevice:
    def __init__(self, name: str):
        self.name = name

class TestMachine:
    def __init__(self, env: TestEnvironment):
        # TODO: implement device detection, matching Blender Benchmark.
        self.devices = [TestDevice('CPU')]

    def cpu_device(self) -> str:
        return self.devices[0]