(self, n: torch.fx.Node)
| 42 | ] = defaultdict(dict) |
| 43 | |
| 44 | def run_node(self, n: torch.fx.Node) -> None: |
| 45 | self.node = n |
| 46 | try: |
| 47 | ret = super().run_node(n) |
| 48 | except Exception as e: |
| 49 | if isinstance(e, (InternalError, ExportError, RunHigherOrderOperatorError)): |
| 50 | raise e |
| 51 | else: |
| 52 | raise InternalError(str(e)) from e |
| 53 | return ret |
| 54 | |
| 55 | def _get_kernel_arg(self, schema_arg, schema_arg_idx, args, kwargs): |
| 56 | if schema_arg.name in kwargs: |
nothing calls this directly
no test coverage detected