Skip to content

Commit

Permalink
Merge branch 'master' into muse-str-inputs-conversion
Browse files Browse the repository at this point in the history
  • Loading branch information
apaniukov authored Feb 6, 2024
2 parents c8a2a7a + 6aa1117 commit 86b56e0
Showing 1 changed file with 0 additions and 17 deletions.
17 changes: 0 additions & 17 deletions modules/nvidia_plugin/tests/functional/core_config.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -2,26 +2,9 @@
// SPDX-License-Identifier: Apache-2.0
//

#include "functional_test_utils/core_config.hpp"

#include "cuda_test_constants.hpp"
#include "shared_test_classes/base/ov_subgraph.hpp"

void CoreConfiguration(LayerTestsUtils::LayerTestsCommon* test) {
std::shared_ptr<InferenceEngine::Core> core = PluginCache::get().ie();
ov::element::Type hint = ov::element::f32;
for (auto& param : test->GetFunction()->get_parameters()) {
if (param->get_output_element_type(0) == ov::element::f16) {
hint = ov::element::f16;
break;
}
}
// Set inference_precision hint to run fp32 model in fp32 runtime precision as default plugin execution precision
// may vary
std::map<std::string, std::string> config = {{ov::hint::inference_precision.name(), hint.get_type_name()}};
core->SetConfig(config, ov::test::utils::DEVICE_NVIDIA);
}

namespace ov {
namespace test {

Expand Down

0 comments on commit 86b56e0

Please sign in to comment.