(x)
| 938 | global_chunk_start = (start // chunk_size) + i |
| 939 | |
| 940 | def _pad(x): |
| 941 | padded = torch.zeros(B, H, curr_padded_len, device=x.device, dtype=x.dtype) |
| 942 | padded[:, :, :curr_seqlen] = x[:, :, start:end] |
| 943 | return padded |
| 944 | |
| 945 | ddA_cs_seq = _pad(ddA_cs) |
| 946 | ddA_cs_rev_seq = _pad(ddA_cs_rev) |
no outgoing calls
no test coverage detected