saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-28 22:56:32 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Temporarily disable two tests for TRT EP (#15578)

Browse source 
We are investigating an issue introduced by TRT 8.6 which causes [TRT EP
CI](https://dev.azure.com/onnxruntime/onnxruntime/_build/results?buildId=967950&view=logs&j=7536d2cd-87d4-54fe-4891-bfbbf2741d83&t=66420422-c7d6-5f71-625c-4b7851c9b9ba)
fail. Disable two tests for now until the issue is root caused and
fixed.
This commit is contained in:
Chi Lo 2023-04-20 16:32:56 -07:00 committed by GitHub
parent dfa06bf81b
commit 6cf080ccbf
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
onnxruntime/test/framework/inference_session_test.cc
View file

@ -1553,7 +1553,7 @@ TEST(InferenceSessionTests, Test3LayerNestedSubgraph) {
so.session_logid = "InferenceSessionTests.Test3LayerNestedSubgraph";
InferenceSession session_object{so, GetEnvironment()};
#if USE_TENSORRT
#if defined(_WIN32) && defined(USE_TENSORRT)
ASSERT_STATUS_OK(session_object.RegisterExecutionProvider(DefaultTensorrtExecutionProvider()));
#elif USE_CUDA
ASSERT_STATUS_OK(session_object.RegisterExecutionProvider(DefaultCudaExecutionProvider()));
@ -1591,7 +1591,7 @@ TEST(InferenceSessionTests, Test3LayerNestedSubgraph) {
ASSERT_TRUE(status.IsOK());
VerifyOutputs(fetches, expected_dims, expected_values);
#if USE_TENSORRT
#if defined(_WIN32) && defined(USE_TENSORRT)
// previous run with graph being optimized, one of If node’s both subgraphs become empty, so TRT EP won’t assign this If node to TRT and later ORT assign it to CUDA.
// we also want to test graph not being optimized and TRT EP should also be able to run it and make the whole graph run on TRT.
so.graph_optimization_level = TransformerLevel::Default;
@ -1705,7 +1705,7 @@ TEST(InferenceSessionTests, Test2LayerNestedSubgraph) {
so.session_logid = "InferenceSessionTests.Test2LayerNestedSubgraph";
InferenceSession session_object{so, GetEnvironment()};
#if USE_TENSORRT
#if defined(_WIN32) && defined(USE_TENSORRT)
ASSERT_STATUS_OK(session_object.RegisterExecutionProvider(DefaultTensorrtExecutionProvider()));
#elif USE_CUDA
ASSERT_STATUS_OK(session_object.RegisterExecutionProvider(DefaultCudaExecutionProvider()));