Skip to content

Tensorrt backend#184

Open
okdshin wants to merge 137 commits intomasterfrom
tensorrt
Open

Tensorrt backend#184
okdshin wants to merge 137 commits intomasterfrom
tensorrt

Conversation

@okdshin
Copy link
Copy Markdown
Contributor

@okdshin okdshin commented Dec 4, 2018

This PR adds TensorRT backend. It enables fast inference by using NVIDIA's TensorRT.

  • Modify CMakeLists.txt for merging
  • Add a license
  • Testing

Comment thread menoh/tensorrt/Inference.cpp Outdated
const int TIMING_ITERATIONS = 1;

typedef std::pair<std::string, float> Record;
std::vector<Record> mProfile;
Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Can't we use std::map<std::string, float> (or unordered version) instead of std::vector<Record> ?

Copy link
Copy Markdown
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thank you comment. I think so too. std::unordered_map seems suitable here.

Comment thread CMakeLists.txt
include(CMakeDependentOption)
option(ENABLE_MKLDNN "Use MKLDNN" ON)
option(ENABLE_TENSORRT "Use TensorRT" OFF)
cmake_dependent_option(ENABLE_TENSORRT_PROFILER "Make TneosrRT Profiler feature available" ON
Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

TneosrRT -> TensorRT

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