diff options
Diffstat (limited to 'tests/InferenceTest.inl')
| -rw-r--r-- | tests/InferenceTest.inl | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tests/InferenceTest.inl b/tests/InferenceTest.inl index 4dde35403d..07a20d5a13 100644 --- a/tests/InferenceTest.inl +++ b/tests/InferenceTest.inl @@ -342,7 +342,7 @@ int ClassifierInferenceTestMain(int argc, } modelParams.m_IsModelBinary = isModelBinary; - modelParams.m_ComputeDevice = modelOptions.m_ComputeDevice; + modelParams.m_ComputeDevices = modelOptions.GetComputeDevicesAsBackendIds(); modelParams.m_VisualizePostOptimizationModel = modelOptions.m_VisualizePostOptimizationModel; modelParams.m_EnableFp16TurboMode = modelOptions.m_EnableFp16TurboMode; |