pytorch

05336c91 - [JIT] Add additional debug output when op cannot be found in AliasDb (#68099)

[JIT] Add additional debug output when op cannot be found in AliasDb (#68099) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/68099 When an op in the graph cannot be matched to any known ops, alias_analysis.cpp throws an error. Before: ``` RuntimeError: 0INTERNAL ASSERT FAILED at "../torch/csrc/jit/ir/alias_analysis.cpp":612, please report a bug to PyTorch. We don't have an op for aten::add but it isn't a special case. Argument types: Tensor, float, Tensor, ``` After: ``` RuntimeError: 0INTERNAL ASSERT FAILED at "../torch/csrc/jit/ir/alias_analysis.cpp":612, please report a bug to PyTorch. We don't have an op for a ten::add but it isn't a special case. Argument types: Tensor, float, Tensor, Candidates: aten::add.Tensor(Tensor self, Tensor other, *, Scalar alpha=1) -> (Tensor) aten::add.Scalar(Tensor self, Scalar other, Scalar alpha=1) -> (Tensor) aten::add.out(Tensor self, Tensor other, *, Scalar alpha=1, Tensor(a!) out) -> (Tensor(a!)) aten::add.t(t[] a, t[] b) -> (t[]) aten::add.str(str a, str b) -> (str) aten::add.int(int a, int b) -> (int) aten::add.complex(complex a, complex b) -> (complex) aten::add.float(float a, float b) -> (float) aten::add.int_complex(int a, complex b) -> (complex) aten::add.complex_int(complex a, int b) -> (complex) aten::add.float_complex(float a, complex b) -> (complex) aten::add.complex_float(complex a, float b) -> (complex) aten::add.int_float(int a, float b) -> (float) aten::add.float_int(float a, int b) -> (float) aten::add(Scalar a, Scalar b) -> (Scalar) ``` Test Plan: Run ``` import torch if __name__ == '__main__': ir = """ graph(%x : Tensor, %y : Tensor): %2 : float = prim::Constant[value=1.2]() %result : Tensor= aten::add(%x, %2, %y) return (%result) """ x = torch.tensor([[1., 2.], [3., 4.]]) y = torch.tensor([[2., 1.], [2., 1.]]) graph = torch._C.parse_ir(ir) print(graph) graph.alias_db().analyze() # print(script(x, y)) ``` to get the results above Imported from OSS Reviewed By: anjali411 Differential Revision: D32339639 fbshipit-source-id: a79a3c2f157154b5fb1e3f33a23e43b7884e8e38