Return a Blockwise-task argument
(self, arg, i)
| 608 | return f"{head}-{self.deterministic_token}" |
| 609 | |
| 610 | def _blockwise_arg(self, arg, i): |
| 611 | """Return a Blockwise-task argument""" |
| 612 | if isinstance(arg, Expr): |
| 613 | # Make key for Expr-based argument |
| 614 | if self._broadcast_dep(arg): |
| 615 | return TaskRef((arg._name, 0)) |
| 616 | else: |
| 617 | return TaskRef((arg._name, i)) |
| 618 | |
| 619 | else: |
| 620 | return arg |
| 621 | |
| 622 | def _task(self, name: Key, index: int) -> Task: |
| 623 | """Produce the task for a specific partition |