(lazy_tensor: LazyTensor, n_part: int)
| 936 | |
| 937 | |
| 938 | def part_lazy(lazy_tensor: LazyTensor, n_part: int) -> LazyTensor: |
| 939 | def load() -> Tensor: |
| 940 | return lazy_tensor.load().part(n_part) |
| 941 | s = lazy_tensor.shape.copy() |
| 942 | s[0] = s[0] // 3 |
| 943 | return LazyTensor(load, s, lazy_tensor.data_type, 'part ' + lazy_tensor.description) |
| 944 | |
| 945 | |
| 946 | def pack_experts_lazy(lazy_tensors: list[LazyTensor]) -> LazyTensor: |
no test coverage detected