(self)
| 549 | |
| 550 | @property |
| 551 | def cum_seqlen(self) -> torch.LongTensor: |
| 552 | cum_seqlen = self.get_spatial_cache('cum_seqlen') |
| 553 | if cum_seqlen is None: |
| 554 | cum_seqlen = torch.cat([ |
| 555 | torch.tensor([0], dtype=torch.long, device=self.device), |
| 556 | self.seqlen.cumsum(dim=0) |
| 557 | ], dim=0) |
| 558 | self.register_spatial_cache('cum_seqlen', cum_seqlen) |
| 559 | return cum_seqlen |
| 560 | |
| 561 | @property |
| 562 | def batch_boardcast_map(self) -> torch.LongTensor: |
nothing calls this directly
no test coverage detected