diff --git a/src/operator/subgraph/tensorrt/tensorrt-inl.h b/src/operator/subgraph/tensorrt/tensorrt-inl.h index cdc8a49e9d13..ca7004064844 100644 --- a/src/operator/subgraph/tensorrt/tensorrt-inl.h +++ b/src/operator/subgraph/tensorrt/tensorrt-inl.h @@ -314,6 +314,8 @@ class TensorrtProperty : public SubgraphProperty { const std::unordered_map& options_map) override { auto it_precision = options_map.find("precision"); if (it_precision != options_map.end()) { + fp16_mode_ = false; + int8_mode_ = false; auto precision_string = it_precision->second; std::replace(precision_string.begin(), precision_string.end(), '_', ' '); std::istringstream iss(precision_string);