-
Notifications
You must be signed in to change notification settings - Fork 3k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Add unit test for TRT EP plugin #22548
Add unit test for TRT EP plugin #22548
Conversation
@@ -0,0 +1,193 @@ | |||
// Copyright (c) Microsoft Corporation. All rights reserved. |
Check warning
Code scanning / lintrunner
CLANGFORMAT/format Warning test
Run lintrunner -a to apply this patch.
samples/utils/common.h
Outdated
@@ -0,0 +1,169 @@ | |||
/** |
Check warning
Code scanning / lintrunner
CLANGFORMAT/format Warning
Run lintrunner -a to apply this patch.
samples/utils/cuda/cuda_call.h
Outdated
@@ -0,0 +1,69 @@ | |||
// Copyright (c) Microsoft Corporation. All rights reserved. |
Check warning
Code scanning / lintrunner
CLANGFORMAT/format Warning
Run lintrunner -a to apply this patch.
samples/utils/exceptions.h
Outdated
@@ -0,0 +1,91 @@ | |||
// Copyright (c) Microsoft Corporation. All rights reserved. |
Check warning
Code scanning / lintrunner
CLANGFORMAT/format Warning
Run lintrunner -a to apply this patch.
Hook TRT EP plugin to run the existing unit test in CI
onnxruntime/test/providers/tensorrt/tensorrt_basic_test.cc
onnxruntime_shared_lib_test
(which links against onnxruntime dll)--test_tensorrt_ep_plugin
to runonnxruntime_shared_lib_test
Note: The unit test doesn't cover all the cases since current TRT EP plugin hasn't added all the features yet, will update later.