MCPcopy Create free account
hub / github.com/pytorch/executorch / EdgeCompileConfig

Class EdgeCompileConfig

exir/capture/_config.py:37–52  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

35@compatibility(is_backward_compatible=False)
36@dataclass
37class 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)

Calls

no outgoing calls