moving the error output to validator

pytorch · Feb 26, 2024 · 13a6f94 · 13a6f94
1 parent 9b24cd6
commit 13a6f94
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/py/torch_tensorrt/dynamo/conversion/ops_evaluators.py b/py/torch_tensorrt/dynamo/conversion/ops_evaluators.py
@@ -106,6 +106,10 @@ def aten_ops_randn(
 def randperm_validator(randperm_node: Node) -> bool:
     dtype = randperm_node.kwargs.get("dtype", None)
     layout = randperm_node.kwargs.get("layout", None)
+    input = randperm_node.args[0]
+    if not isinstance(input, int):
+        _LOGGER.error(f"Input should be of type int.")
+        return False
     if dtype is not None:
         _LOGGER.debug(
             f"Currently we don't support specifying output dtype, got {dtype}."