(config, quantization_model_name, origin_model_name, quantization_type, cuda_prefix: str = '')
| 37 | |
| 38 | |
| 39 | def quantization_all(config, quantization_model_name, origin_model_name, quantization_type, cuda_prefix: str = ''): |
| 40 | result, msg = quantization(config, quantization_model_name, origin_model_name, quantization_type, cuda_prefix) |
| 41 | log_path = config.get('log_path') |
| 42 | quantization_log = os.path.join( |
| 43 | log_path, '_'.join(['quantization', quantization_type, |
| 44 | quantization_model_name.split('/')[1]]) + '.log') |
| 45 | |
| 46 | allure.attach.file(quantization_log, name=quantization_log, attachment_type=allure.attachment_type.TEXT) |
| 47 | assert result, msg |
no test coverage detected