Skip to content
Snippets Groups Projects
Commit 7f1ab7e5 authored by Michael Kuron's avatar Michael Kuron :mortar_board:
Browse files

more test fix

parent e8319d7c
No related merge requests found
......@@ -79,6 +79,7 @@ ubuntu:
- pytest-3 -v -m "not longrun"
tags:
- docker
- cuda
minimal-conda:
stage: test
......
......@@ -107,9 +107,11 @@ def memory_sizes_of_current_machine():
if get_cpu_info:
cpu_info = get_cpu_info()
result.update({'L1': cpu_info['l1_data_cache_size'],
'L2': cpu_info['l2_cache_size'],
'L3': cpu_info['l3_cache_size']})
if 'l1_data_cache_size' in cpu_info:
result['L1'] = cpu_info['l1_data_cache_size']
result['L2'] = cpu_info['l2_cache_size']
if 'l3_cache_size' in cpu_info:
result['L3'] = cpu_info['l3_cache_size']
if device:
size = device.total_memory() / (1024 * 1024)
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment