diff options
| author | Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> | 2019-01-31 16:44:26 +0000 |
|---|---|---|
| committer | Matteo Martincigh <matteo.martincigh@arm.com> | 2019-02-01 09:10:51 +0000 |
| commit | 339bcae73515c66899432b5844d7c239c570c4b8 (patch) | |
| tree | 91bea3fc1eda5cf96309ad2a255917667ffad679 /tests/TfLiteMobileNetSsd-Armnn | |
| parent | 0c051f9b6b8d7d1602e81d2977dda449b6392642 (diff) | |
| download | armnn-339bcae73515c66899432b5844d7c239c570c4b8.tar.gz | |
IVGCVSW-2604 Fix bug that made it impossible to execute inference tests on certain backends
* Read compute devices from the CL as strings and convert them into BackendId objects afterwards
Change-Id: Icded1c572778f5a213644e3052ff6dfe7022128b
Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Diffstat (limited to 'tests/TfLiteMobileNetSsd-Armnn')
| -rw-r--r-- | tests/TfLiteMobileNetSsd-Armnn/TfLiteMobileNetSsd-Armnn.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tests/TfLiteMobileNetSsd-Armnn/TfLiteMobileNetSsd-Armnn.cpp b/tests/TfLiteMobileNetSsd-Armnn/TfLiteMobileNetSsd-Armnn.cpp index b1bc0f6120..3328339318 100644 --- a/tests/TfLiteMobileNetSsd-Armnn/TfLiteMobileNetSsd-Armnn.cpp +++ b/tests/TfLiteMobileNetSsd-Armnn/TfLiteMobileNetSsd-Armnn.cpp @@ -59,7 +59,7 @@ int main(int argc, char* argv[]) modelParams.m_InputShapes = { inputTensorShape }; modelParams.m_IsModelBinary = true; - modelParams.m_ComputeDevice = modelOptions.m_ComputeDevice; + modelParams.m_ComputeDevices = modelOptions.GetComputeDevicesAsBackendIds(); modelParams.m_VisualizePostOptimizationModel = modelOptions.m_VisualizePostOptimizationModel; modelParams.m_EnableFp16TurboMode = modelOptions.m_EnableFp16TurboMode; |