blackteay/ffmpeg-python
1
0
Fork 0
mirror of https://github.com/kkroening/ffmpeg-python.git synced 2025-08-10 13:10:03 +08:00
Code Issues Projects Releases Wiki Activity

Detect: Integrate GPU codecs data into the GPU detection

Browse Source
This commit is contained in:
Ross Patterson 2019-08-10 18:39:21 -07:00
parent 054aadb56c
commit d3ac1865a4
1 changed files with 18 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
ffmpeg/_detect.py
View File

@ -117,6 +117,7 @@ def detect_gpus():
for gpu in gpus: for gpu in gpus:
vendor_data = data.get(gpu.get('vendor', '').lower()) vendor_data = data.get(gpu.get('vendor', '').lower())
if vendor_data: if vendor_data:
model_lines_data = _parse_models( model_lines_data = _parse_models(
model_lines=vendor_data['lines'], model_lines=vendor_data['lines'],
boards=gpu['board'].lower(), model_data={}) boards=gpu['board'].lower(), model_data={})
@ -124,6 +125,23 @@ def detect_gpus():
gpu['model_num'] = list(model_lines_data[ gpu['model_num'] = list(model_lines_data[
gpu['model_line']]['models'].keys())[0] gpu['model_line']]['models'].keys())[0]
for coder_type in ['encoders', 'decoders']:
model_line_data = vendor_data[coder_type]['model_lines'][
gpu['model_line']]
coder_boards = model_line_data['models'].get(
gpu['model_num'])
if coder_boards is None:
for model_range, boards in model_line_data[
'model_ranges']:
# TODO proper model range matching
if gpu['model_num'] in model_range:
coder_boards = boards
break
if coder_boards is None:
continue
gpu[coder_type] = vendor_data[coder_type]['boards'][
coder_boards]
return gpus return gpus