(cls, ep, head)
| 562 | |
| 563 | @classmethod |
| 564 | def maybe_create(cls, ep, head): |
| 565 | if not match_target(head, torch.ops.aten.embedding.default): |
| 566 | return None |
| 567 | |
| 568 | w, x = head.args[0:2] |
| 569 | if not isinstance(w, Node): |
| 570 | return None |
| 571 | if not has_single_user(w): |
| 572 | return None |
| 573 | parsed = parse_dequant_nvfp4_node(w) |
| 574 | if parsed is None: |
| 575 | return None |
| 576 | qdata, scale, per_tensor_scale, output_dtype = parsed |
| 577 | return cls(head, [w], qdata, scale, per_tensor_scale, output_dtype) |
| 578 | |
| 579 | def __call__(self, P: MLXProgramBuilder, n: Node) -> Slot: |
| 580 | assert n == self.head |
nothing calls this directly
no test coverage detected