[incubator-mxnet] branch v1.9.x updated: backport #20277

[sa...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

samskalicky pushed a commit to branch v1.9.x
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git


The following commit(s) were added to refs/heads/v1.9.x by this push:
     new 7ce251e  backport #20277
7ce251e is described below

commit 7ce251e61798b3bd97d9c5d239b76be031868858
Author: Serge Panev <sp...@nvidia.com>
AuthorDate: Sat May 22 00:37:20 2021 +0900

    backport #20277
    
    Signed-off-by: Serge Panev <sp...@nvidia.com>
---
 src/operator/subgraph/tensorrt/tensorrt-inl.h | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/operator/subgraph/tensorrt/tensorrt-inl.h b/src/operator/subgraph/tensorrt/tensorrt-inl.h
index cdc8a49..4258e76 100644
--- a/src/operator/subgraph/tensorrt/tensorrt-inl.h
+++ b/src/operator/subgraph/tensorrt/tensorrt-inl.h
@@ -312,6 +312,8 @@ class TensorrtProperty : public SubgraphProperty {
 
   void PrePartition(const nnvm::Graph& g,
     const std::unordered_map<std::string, std::string>& options_map) override {
+    fp16_mode_ = false;
+    int8_mode_ = false;
     auto it_precision = options_map.find("precision");
     if (it_precision != options_map.end()) {
       auto precision_string = it_precision->second;