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

[ONNXRuntime] add tensorRT option #1611

Merged
merged 2 commits into from
Apr 29, 2022
Merged

Conversation

lanking520
Copy link
Contributor

Description

Allow user to specify TensorRT mode under GPU.

@codecov-commenter
Copy link

codecov-commenter commented Apr 28, 2022

Codecov Report

Merging #1611 (ed623ef) into master (bb5073f) will decrease coverage by 1.31%.
The diff coverage is 59.11%.

@@             Coverage Diff              @@
##             master    #1611      +/-   ##
============================================
- Coverage     72.08%   70.77%   -1.32%     
- Complexity     5126     5400     +274     
============================================
  Files           473      504      +31     
  Lines         21970    23642    +1672     
  Branches       2351     2576     +225     
============================================
+ Hits          15838    16732     +894     
- Misses         4925     5625     +700     
- Partials       1207     1285      +78     
Impacted Files Coverage Δ
api/src/main/java/ai/djl/modality/cv/Image.java 69.23% <ø> (-4.11%) ⬇️
...i/djl/modality/cv/translator/BigGANTranslator.java 21.42% <ø> (-5.24%) ⬇️
...odality/cv/translator/BigGANTranslatorFactory.java 33.33% <0.00%> (+8.33%) ⬆️
...nslator/InstanceSegmentationTranslatorFactory.java 14.28% <0.00%> (-3.90%) ⬇️
.../modality/cv/translator/YoloTranslatorFactory.java 8.33% <0.00%> (-1.67%) ⬇️
...i/djl/modality/cv/translator/YoloV5Translator.java 5.69% <0.00%> (ø)
...odality/cv/translator/YoloV5TranslatorFactory.java 8.33% <0.00%> (-1.67%) ⬇️
...pi/src/main/java/ai/djl/ndarray/BytesSupplier.java 54.54% <0.00%> (-12.13%) ⬇️
...ain/java/ai/djl/ndarray/index/dim/NDIndexPick.java 100.00% <ø> (ø)
api/src/main/java/ai/djl/nn/Blocks.java 75.00% <0.00%> (-25.00%) ⬇️
... and 233 more

Continue to review full report at Codecov.

Legend - Click here to learn more
Δ = absolute <relative> (impact), ø = not affected, ? = missing data
Powered by Codecov. Last update 009ca1a...ed623ef. Read the comment docs.

Change-Id: I5812a0621b3d5019e3cf7a873e11641cd7dd5658
@lanking520 lanking520 merged commit 260a08d into deepjavalibrary:master Apr 29, 2022
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.

3 participants