Skip to content

Commit

Permalink
[TRT][BYOC] handling dynamism in TensorRT to support OD models (apach…
Browse files Browse the repository at this point in the history
…e#6905)

* handling dynamism in TensorRT to support OD models

refactoring test tensort code

added comments to dynamic check wrapper

log.warn changed to logger.info

TRT codegen taking slice_mode into account

TRT codegen to handle both stride_mode

refactoring TRT codegen

adding a test for dynamic offload

[TRT] bug in codegen for slice_mode=end

ctx determined from target in test + io test was missing

* Addressed the formatting/refactoring comments

* Addressed comment in TRT codegen

Lint formatting

* Lint error

* using slice_mode during strided slice registration in tensorrt.py

* removed a few blank lines

* addressing cli comment on elif-return

* Added decorator for tensorrt functions with dynamism check

skip_codegen added for test_tensorrt::test_dynamic_offload

* addressed comments in PR + black linting

* resolved import error in test_tensorrt

* import mxnet location changed to pass CI

* test_integration removed as components were run by pytest anyway
  • Loading branch information
rohanmukh authored and Trevor Morris committed Dec 4, 2020
1 parent fcf9e78 commit c5aebfa
Show file tree
Hide file tree
Showing 4 changed files with 438 additions and 244 deletions.
Loading

0 comments on commit c5aebfa

Please sign in to comment.