-
Notifications
You must be signed in to change notification settings - Fork 369
Open
Labels
Blocked [PyTorch]Issue is blocked by some limitation of PyTorchIssue is blocked by some limitation of PyTorch
Description
- Remove class override of
ConstantFolder
once Torch addresses the issue - Issue filed here: [Inductor]
ConstantFolder
Utility Breaking in Recent Nightly pytorch#110904
To remove:
TensorRT/py/torch_tensorrt/dynamo/lowering/passes/constant_folding.py
Lines 63 to 71 in 0e4c5d8
# TODO: Delete this class when the following code is fixed in nightly: | |
# https://github.com/pytorch/pytorch/blob/4b881b0da390c1290bb12850ef9daad6f6eb2cb6/torch/_inductor/constant_folding.py#L53-L63 | |
class _TorchTensorRTConstantFolder(ConstantFolder): # type: ignore[misc] | |
def __init__(self, *args: Any, **kwargs: Any) -> None: | |
super().__init__(*args, **kwargs) | |
# TODO: Update this function when quantization is added | |
def is_impure(self, node: torch.fx.node.Node) -> bool: | |
return False |
Metadata
Metadata
Assignees
Labels
Blocked [PyTorch]Issue is blocked by some limitation of PyTorchIssue is blocked by some limitation of PyTorch