Skip to content
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

Merged
merged 10 commits into from
Nov 2, 2024

Conversation

chilo-ms
Copy link
Contributor

@chilo-ms chilo-ms commented Oct 22, 2024

Hook TRT EP plugin to run the existing unit test in CI

  • Migrate from onnxruntime/test/providers/tensorrt/tensorrt_basic_test.cc
  • Replace internal APIs with new EP APIs
  • Add unit test in onnxruntime_shared_lib_test (which links against onnxruntime dll)
  • Build ORT with --test_tensorrt_ep_plugin to run onnxruntime_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.

@chilo-ms chilo-ms requested a review from jslhcl October 22, 2024 22:41
@@ -0,0 +1,193 @@
// Copyright (c) Microsoft Corporation. All rights reserved.

Check warning

Code scanning / lintrunner

CLANGFORMAT/format Warning test

See https://clang.llvm.org/docs/ClangFormat.html.
Run lintrunner -a to apply this patch.
@@ -0,0 +1,169 @@
/**

Check warning

Code scanning / lintrunner

CLANGFORMAT/format Warning

See https://clang.llvm.org/docs/ClangFormat.html.
Run lintrunner -a to apply this patch.
@@ -0,0 +1,69 @@
// Copyright (c) Microsoft Corporation. All rights reserved.

Check warning

Code scanning / lintrunner

CLANGFORMAT/format Warning

See https://clang.llvm.org/docs/ClangFormat.html.
Run lintrunner -a to apply this patch.
@@ -0,0 +1,91 @@
// Copyright (c) Microsoft Corporation. All rights reserved.

Check warning

Code scanning / lintrunner

CLANGFORMAT/format Warning

See https://clang.llvm.org/docs/ClangFormat.html.
Run lintrunner -a to apply this patch.
@chilo-ms chilo-ms merged commit 08e3f20 into leca/outOfTreeEP Nov 2, 2024
31 of 36 checks passed
@chilo-ms chilo-ms deleted the chi/add_unit_test_for_trt_ep_plugin branch November 2, 2024 17:33
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant