(batch_devs:list[Compiled], new_call:UOp)
| 162 | |
| 163 | @staticmethod |
| 164 | def _all_devs(batch_devs:list[Compiled], new_call:UOp) -> list[Compiled]: |
| 165 | return dedup(batch_devs + [Device[x] for b in get_call_arg_uops(new_call) |
| 166 | for x in (b.device if isinstance(b.device, tuple) else (b.device,))]) |
| 167 | |
| 168 | @staticmethod |
| 169 | def supports_uop(batch_devs:list[Compiled], new_call:UOp) -> bool: |
no test coverage detected