sparsify.utils package

Submodules

sparsify.utils.system module

Utilities for ONNX based ML system info and validation

sparsify.utils.system.available_ml_engines()List[str][source]
Returns

List of available inference providers on current system. Potential values include [‘deepsparse’, ‘ort_cpu’, ‘ort_gpu’]

sparsify.utils.system.get_ml_sys_info()Dict[str, Any][source]
Returns

a dictionary containing info for the system and ML engines on the system. Such as number of cores, instruction sets, available engines, etc

sparsify.utils.system.ml_engines_errors()Dict[str, Exception][source]
Returns

a dictionary containing any errors encountered when importing ML engines on the current system

Module contents

Utility functions for sparsify