Skip to content

Commit 4b18210

Browse files
authored
fix build break when multi EP is enabled (inference_session_test.cc) (microsoft#25329)
1 parent 3557420 commit 4b18210

File tree

1 file changed

+19
-13
lines changed

1 file changed

+19
-13
lines changed

onnxruntime/test/framework/inference_session_test.cc

Lines changed: 19 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -1134,23 +1134,29 @@ static void TestBindHelper(const std::string& log_str,
11341134

11351135
if (bind_provider_type == kCudaExecutionProvider || bind_provider_type == kRocmExecutionProvider || bind_provider_type == kWebGpuExecutionProvider) {
11361136
#ifdef USE_CUDA
1137-
auto provider = DefaultCudaExecutionProvider();
1138-
gpu_provider = provider.get();
1139-
ASSERT_STATUS_OK(session_object.RegisterExecutionProvider(std::move(provider)));
1137+
{
1138+
auto provider = DefaultCudaExecutionProvider();
1139+
gpu_provider = provider.get();
1140+
ASSERT_STATUS_OK(session_object.RegisterExecutionProvider(std::move(provider)));
1141+
}
11401142
#endif
11411143
#ifdef USE_ROCM
1142-
auto provider = DefaultRocmExecutionProvider();
1143-
gpu_provider = provider.get();
1144-
ASSERT_STATUS_OK(session_object.RegisterExecutionProvider(std::move(provider)));
1144+
{
1145+
auto provider = DefaultRocmExecutionProvider();
1146+
gpu_provider = provider.get();
1147+
ASSERT_STATUS_OK(session_object.RegisterExecutionProvider(std::move(provider)));
1148+
}
11451149
#endif
11461150
#ifdef USE_WEBGPU
1147-
ConfigOptions config_options{};
1148-
ORT_ENFORCE(config_options.AddConfigEntry(webgpu::options::kEnableGraphCapture,
1149-
enable_graph_capture ? webgpu::options::kEnableGraphCapture_ON : webgpu::options::kEnableGraphCapture_OFF)
1150-
.IsOK());
1151-
auto provider = WebGpuExecutionProviderWithOptions(config_options);
1152-
gpu_provider = provider.get();
1153-
ASSERT_STATUS_OK(session_object.RegisterExecutionProvider(std::move(provider)));
1151+
{
1152+
ConfigOptions config_options{};
1153+
ORT_ENFORCE(config_options.AddConfigEntry(webgpu::options::kEnableGraphCapture,
1154+
enable_graph_capture ? webgpu::options::kEnableGraphCapture_ON : webgpu::options::kEnableGraphCapture_OFF)
1155+
.IsOK());
1156+
auto provider = WebGpuExecutionProviderWithOptions(config_options);
1157+
gpu_provider = provider.get();
1158+
ASSERT_STATUS_OK(session_object.RegisterExecutionProvider(std::move(provider)));
1159+
}
11541160
#endif
11551161
}
11561162

0 commit comments

Comments
 (0)