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 cbcea33 commit 647a526Copy full SHA for 647a526
tensorrt_llm/_torch/auto_deploy/utils/pattern_matcher.py
@@ -43,11 +43,13 @@ def _patch_unsupported_input_tensor():
43
"""
44
original_fn = lowering.unsupported_input_tensor
45
46
- def patched_fn(t: torch.Tensor, parent=None, node=None):
+ def patched_fn(t: torch.Tensor, *args, **kwargs):
47
"""Bypass meta tensor check."""
48
if t.is_meta:
49
return False
50
- return original_fn(t, parent, node)
+ return original_fn(
51
+ t, *args, **kwargs
52
+ ) # a generic pass-through of the arguments to accommodate torch side change
53
54
lowering.unsupported_input_tensor = patched_fn
55
try:
0 commit comments