diff --git a/docs/v1.1.0/._index.html b/docs/v1.1.0/._index.html deleted file mode 100644 index b5dadab695..0000000000 Binary files a/docs/v1.1.0/._index.html and /dev/null differ diff --git a/py/torch_tensorrt/fx/tracer/acc_tracer/acc_ops.py b/py/torch_tensorrt/fx/tracer/acc_tracer/acc_ops.py index b28bf263c2..1ce6fa39ea 100644 --- a/py/torch_tensorrt/fx/tracer/acc_tracer/acc_ops.py +++ b/py/torch_tensorrt/fx/tracer/acc_tracer/acc_ops.py @@ -739,6 +739,13 @@ def square_mapper(node: torch.fx.Node, _: nn.Module) -> torch.fx.Node: ("mat2", "other"), ], ) +@register_acc_op_mapping( + op_and_target=("call_function", torch.mm), + arg_replacement_tuples=[ + ("input", "input"), + ("mat2", "other"), + ], +) @register_acc_op_mapping(op_and_target=("call_function", torch.matmul)) @register_acc_op def matmul(*, input, other):