Implement compute device selector
This commit is contained in:
@@ -162,6 +162,12 @@ def main():
|
||||
scene.cycles.samples = 1
|
||||
scene.cycles.aa_samples = 1
|
||||
|
||||
# Cycles device.
|
||||
device = 'CPU' if args.benchmark_device_type == 'CPU' else 'GPU'
|
||||
for scene in bpy.data.scenes:
|
||||
scene.cycles.device = device
|
||||
# TODO(sergey): Tweak tile size.
|
||||
|
||||
# Report number of CPU threads used.
|
||||
for scene in bpy.data.scenes:
|
||||
print("CPU threads used: {}" . format(scene.render.threads))
|
||||
|
Reference in New Issue
Block a user