backport #20277

Signed-off-by: Serge Panev <spanev@nvidia.com>
apache · May 21, 2021 · 7ce251e · 7ce251e
1 parent 22a8308
commit 7ce251e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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;