Files
blender/tests/performance/api/device.py
Brecht Van Lommel dc3f46d96b Tests: performance testing framework
These are scripts for benchmarking Blender features on real-world .blend
files. They were originally written for benchmarking Cycles performance, and
were made generic so they can be used for more Blender features.

The benchmarks can be run locally by developers. But the plan is to also run
these as part of continuous integration to track performance over time.

Currently there are tests for Cycles rendering and .blend file loading.

Documentation:
https://wiki.blender.org/wiki/Tools/Tests/Performance

Main features:
* User created configurations to quickly run, re-run and analyze a selected
  subset of tests.
* Supports both benchmarking with existing builds, and automatic building of
  specified git commits, tags and branches.
* Generate HTML page with bar and line graphs from test results.
* Controlled using simple command line tool.
* For writing tests, convenient abstraction to run a Python function in Blender
  with arguments and return value.

Ref T74730

Differential Revision: https://developer.blender.org/D11662
2021-07-05 12:32:32 +02:00

69 lines
2.2 KiB
Python

# Apache License, Version 2.0
import platform
import subprocess
from typing import List
def get_cpu_name() -> str:
# Get full CPU name.
if platform.system() == "Windows":
return platform.processor()
elif platform.system() == "Darwin":
cmd = ['/usr/sbin/sysctl', "-n", "machdep.cpu.brand_string"]
return subprocess.check_output(cmd).strip().decode('utf-8')
else:
with open('/proc/cpuinfo') as f:
for line in f:
if line.startswith('model name'):
return line.split(':')[1].strip()
return "Unknown CPU"
def get_gpu_device(args: None) -> List:
# Get the list of available Cycles GPU devices.
import bpy
import sys
prefs = bpy.context.preferences
cprefs = prefs.addons['cycles'].preferences
result = []
for device_type, _, _, _ in cprefs.get_device_types(bpy.context):
cprefs.compute_device_type = device_type
devices = cprefs.get_devices_for_type(device_type)
index = 0
for device in devices:
if device.type == device_type:
result.append({'type': device.type, 'name': device.name, 'index': index})
index += 1
break
return result
class TestDevice:
def __init__(self, device_type: str, device_id: str, name: str, operating_system: str):
self.type = device_type
self.id = device_id
self.name = name
self.operating_system = operating_system
class TestMachine:
def __init__(self, env, need_gpus: bool):
operating_system = platform.system()
self.devices = [TestDevice('CPU', 'CPU', get_cpu_name(), operating_system)]
self.has_gpus = need_gpus
if need_gpus and env.blender_executable:
gpu_devices, _ = env.run_in_blender(get_gpu_device, {})
for gpu_device in gpu_devices:
device_type = gpu_device['type']
device_name = gpu_device['name']
device_id = gpu_device['type'] + "_" + str(gpu_device['index'])
self.devices.append(TestDevice(device_type, device_id, device_name, operating_system))
def cpu_device(self) -> TestDevice:
return self.devices[0]