Fix bug check trt (#10600)

Co-authored-by: Michalis Papapdimitriou <mpapapdimitriou@octoml.ai>
apache · Mar 15, 2022 · 0b61256 · 0b61256
1 parent 2f7bb58
commit 0b61256
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/python/tvm/relay/op/contrib/tensorrt.py b/python/tvm/relay/op/contrib/tensorrt.py
@@ -38,10 +38,10 @@ def is_supported_trt_dtype(args):
     ret: bool
         True if supported, False if not.
     """
-    if any([x.checked_type.dtype in supported_types for x in args]):
+    if not all([x.checked_type.dtype in supported_types for x in args]):
         logger.info("Only float32 and float16 inputs are supported for TensorRT BYOC.")
-        return True
-    return False
+        return False
+    return True
 
 
 def is_tensorrt_runtime_enabled():