Fix bug check trt (apache#10600)

Co-authored-by: Michalis Papapdimitriou <mpapapdimitriou@octoml.ai>
pfk-beta · Apr 11, 2022 · 71b95ba · 71b95ba
1 parent ee516a8
commit 71b95ba
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():