TristonC opened a new pull request #18450: URL: https://github.com/apache/incubator-mxnet/pull/18450
## Description ## It enables build MXNet on Jetson with TensorRT support. Without this, the TensorRT integration source code under mxnet/src/operator/subgraph/tensorrt will not be correctly compiled. ## Checklist ## ### Essentials ### Please feel free to remove inapplicable items for your PR. - - [x] The PR title starts with [MXNET-1451], where $JIRA_ID refers to the relevant [JIRA issue](https://issues.apache.org/jira/projects/MXNET/issues) created (except PRs with tiny changes) - [x] Changes are complete (i.e. I finished coding on this PR) - [x] To the best of my knowledge, examples are either not affected by this change, or have been fixed to be compatible with this change ### Changes ### - Turn USE_TENSRORT option on for cmake to enable tensorrt source files under src/operator/subgraph/tenssorrt - Change the CUDA arch from 5.2 to 5.3 ## Comments ## - This change should only affect MXNet jetson wheels. Current jetson wheel build is done on only one GPU arch. ---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: [email protected]
