| 102 | |
| 103 | |
| 104 | class TasksRechunk(Rechunk): |
| 105 | _parameters = ["array", "_chunks", "threshold", "block_size_limit"] |
| 106 | |
| 107 | @cached_property |
| 108 | def chunks(self): |
| 109 | return self.operand("_chunks") |
| 110 | |
| 111 | def _lower(self): |
| 112 | return |
| 113 | |
| 114 | def _layer(self): |
| 115 | steps = plan_rechunk( |
| 116 | self.array.chunks, |
| 117 | self.chunks, |
| 118 | self.array.dtype.itemsize, |
| 119 | self.threshold, |
| 120 | self.block_size_limit, |
| 121 | ) |
| 122 | name = self.array.name |
| 123 | old_chunks = self.array.chunks |
| 124 | layers = [] |
| 125 | for i, c in enumerate(steps): |
| 126 | level = len(steps) - i - 1 |
| 127 | name, old_chunks, layer = _compute_rechunk( |
| 128 | name, old_chunks, c, level, self.name |
| 129 | ) |
| 130 | layers.append(layer) |
| 131 | |
| 132 | return toolz.merge(*layers) |
| 133 | |
| 134 | |
| 135 | def _compute_rechunk(old_name, old_chunks, chunks, level, name): |
no outgoing calls
no test coverage detected
searching dependent graphs…