We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent ee09763 commit 9c1569eCopy full SHA for 9c1569e
py/torch_tensorrt/dynamo/conversion/ops_evaluators.py
@@ -106,6 +106,10 @@ def aten_ops_randn(
106
def randperm_validator(randperm_node: Node) -> bool:
107
dtype = randperm_node.kwargs.get("dtype", None)
108
layout = randperm_node.kwargs.get("layout", None)
109
+ input = randperm_node.args[0]
110
+ if not isinstance(input, int):
111
+ _LOGGER.error(f"Input should be of type int.")
112
+ return False
113
if dtype is not None:
114
_LOGGER.debug(
115
f"Currently we don't support specifying output dtype, got {dtype}."
0 commit comments