Coverage for ibllib/misc/misc.py: 50%

1# library of small functions 

2import logging (empty)

3import subprocess (empty)

4 

5import numpy as np (empty)

6 

7from ibllib.exceptions import NvidiaDriverNotReady (empty)

8 

9_logger = logging.getLogger(__name__) (empty)

10 

11 

12def _parametrized(dec): (empty)

13 def layer(*args, **kwargs): 

14 def repl(f): 

15 return dec(f, *args, **kwargs) 

16 return repl 

17 return layer 

18 

19 

20def structarr(names, shape=None, formats=None): (empty)

21 if not formats: 5 ctx1bcdef

22 formats = ['f8'] * len(names) 

23 dtyp = np.dtype({'names': names, 'formats': formats}) 5 ctx1bcdef

24 return np.zeros(shape, dtype=dtyp) 5 ctx1bcdef

25 

26 

27def check_nvidia_driver(): (empty)

28 """ 

29 Checks if the GPU driver reacts and otherwise raises a custom error. 

30 Useful to check before long GPU-dependent processes. 

31 """ 

32 process = subprocess.Popen('nvidia-smi', shell=True, stdout=subprocess.PIPE, 

33 stderr=subprocess.PIPE, executable="/bin/bash") 

34 info, error = process.communicate() 

35 if process.returncode != 0: 

36 raise NvidiaDriverNotReady(f"{error.decode('utf-8')}") 

37 _logger.info("nvidia-smi command successful")