(self)
| 159 | |
| 160 | class CompositeDelegateModule(torch.nn.Module): |
| 161 | def __init__(self) -> None: |
| 162 | super().__init__() |
| 163 | |
| 164 | class DelegateAdd(nn.Module): |
| 165 | def __init__(self) -> None: |
| 166 | super().__init__() |
| 167 | |
| 168 | def forward(self, x: Tensor, y: Tensor) -> Tensor: |
| 169 | return [x + y] |
| 170 | |
| 171 | def get_random_inputs(self) -> Tuple[Tensor, Tensor]: |
| 172 | return (torch.randn(1, 3), torch.randn(1, 3)) |
| 173 | |
| 174 | delegated_m = DelegateAdd() |
| 175 | edge_ir_m = to_edge( |
| 176 | export(delegated_m, delegated_m.get_random_inputs(), strict=True) |
| 177 | ) |
| 178 | lowered_module = LoweredBackendModule( |
| 179 | edge_program=edge_ir_m.exported_program(), |
| 180 | backend_id="backend_demo", |
| 181 | processed_bytes=bytes("basic_module_add", encoding="utf8"), |
| 182 | compile_specs=[], |
| 183 | ) |
| 184 | self.lowered_module: LoweredBackendModule = lowered_module |
| 185 | |
| 186 | def forward(self, a: exir.Value, b: exir.Value, s: Tensor) -> Tensor: |
| 187 | res = self.lowered_module(a, b) |
nothing calls this directly
no test coverage detected