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