| 35 | @compatibility(is_backward_compatible=False) |
| 36 | @dataclass |
| 37 | class EdgeCompileConfig: |
| 38 | # TODO(qihan): remove ability to opt out |
| 39 | _check_ir_validity: bool = True |
| 40 | # TODO(larryliu): remove this |
| 41 | _use_edge_ops: bool = True |
| 42 | # TODO(gasoonjia): remove this |
| 43 | _skip_dim_order: bool = False |
| 44 | # Allow core ATen ops check to be skipped for certain ops, but continue with the rest of the checks. |
| 45 | # Note: only use this for core ATen ops that are missing decompositions. This is temporary, |
| 46 | # enabling verification on the rest of the program until decomposition coverage is improved. |
| 47 | _core_aten_ops_exception_list: List[torch._ops.OpOverload] = field( |
| 48 | default_factory=list |
| 49 | ) |
| 50 | # Allow ops to be preserved in the graph, i.e., prevent them from being decomposed. |
| 51 | # These may be core or non-core ATen ops; custom ops should not be here. |
| 52 | preserve_ops: List[torch.torch._ops.OpOverload] = field(default_factory=list) |
| 53 | |
| 54 | |
| 55 | @compatibility(is_backward_compatible=False) |
no outgoing calls