diff options
| author | Ryan OShea <ryan.oshea3@arm.com> | 2022-11-07 16:20:48 +0000 |
|---|---|---|
| committer | ryan.oshea3 <ryan.oshea3@arm.com> | 2022-11-16 15:22:50 +0000 |
| commit | 31441595009182c985dacbedc70c41ee6664d070 (patch) | |
| tree | 248a85295aeff4022c9b395fc97748b0a0aa6b35 /delegate/python | |
| parent | bd18eab07a8f30492de1e462b1815189014cb8d5 (diff) | |
| download | armnn-31441595009182c985dacbedc70c41ee6664d070.tar.gz | |
IVGCVSW-7214 Disable BF16-Turbo-Mode and remove conversion layers
- Remove Bf16ToFp32 Conversion Layer
- Remove Fp32ToBf16 Conversion Layer
- Remove B16 Conversion tests
* Throw exception if m_ReduceFp32ToBf16 optimzer option is set to true
* Provide comments to enable fast math in order to use bf16
* Update docs to inform users to enable fast math for bf16
Execute Network Changes
* Require bf16_turbo_mode to also have fast_math_enabled set to true
- Remove setting m_ReduceFp32ToBf16 optimizer option
Signed-off-by: Ryan OShea <ryan.oshea3@arm.com>
Change-Id: Ibaa6da9d29c96a1ce32ff5196b0847fde9f04a1c
Diffstat (limited to 'delegate/python')
| -rw-r--r-- | delegate/python/test/test_external_delegate.py | 33 |
1 files changed, 0 insertions, 33 deletions
diff --git a/delegate/python/test/test_external_delegate.py b/delegate/python/test/test_external_delegate.py index a8dd8e6d3e..fe58d57e90 100644 --- a/delegate/python/test/test_external_delegate.py +++ b/delegate/python/test/test_external_delegate.py @@ -218,39 +218,6 @@ def test_external_delegate_options_fp32_to_fp16(capfd, delegate_dir, test_data_f assert "convert_fp32_to_fp16" in captured.out assert "convert_fp16_to_fp32" in captured.out -def test_external_delegate_options_fp32_to_bf16(capfd, delegate_dir, test_data_folder): - # create armnn delegate with reduce-fp32-to-bf16 option - armnn_delegate = tflite.load_delegate(delegate_dir, options = {"backends": "CpuRef", - "debug-data": "1", - "reduce-fp32-to-bf16": "1"}) - - model_file_name = "conv2d.tflite" - - inputShape = [ 1, 5, 5, 1 ] - outputShape = [ 1, 3, 3, 1 ] - - inputValues = [ 1, 5, 2, 3, 5, - 8, 7, 3, 6, 3, - 3, 3, 9, 1, 9, - 4, 1, 8, 1, 3, - 6, 8, 1, 9, 2 ] - - expectedResult = [ 28, 38, 29, - 96, 104, 53, - 31, 55, 24 ] - - input = np.array(inputValues, dtype=np.float32).reshape(inputShape) - expected_output = np.array(expectedResult, dtype=np.float32).reshape(outputShape) - - # run the inference - armnn_outputs = run_inference(test_data_folder, model_file_name, [input], [armnn_delegate]) - - # check results - compare_outputs(armnn_outputs, [expected_output]) - - captured = capfd.readouterr() - assert "convert_fp32_to_bf16" in captured.out - def test_external_delegate_options_memory_import(delegate_dir, test_data_folder): # create armnn delegate with memory-import option armnn_delegate = tflite.load_delegate(delegate_dir, options = {"backends": "CpuAcc,CpuRef", |