test_hetr_integration.py 文件源码

def test_hetr_benchmark(hetr_device, config):
    pytest.skip('Possible issue only on jenkins, disable until figured out.')
    """
    Description:
        Test to ensure benchmarks are working.
        Benchmark used for test is mini_resnet
    """
    from examples.benchmarks.mini_resnet import run_resnet_benchmark
    c = config
    run_resnet_benchmark(dataset=c['dataset'],
                         num_iterations=c['iter_count'],
                         n_skip=1,
                         batch_size=c['batch_size'],
                         device_id=c['device_id'],
                         transformer_type='hetr',
                         device=hetr_device,
                         bprop=c['bprop'],
                         batch_norm=c['batch_norm'],
                         visualize=False)