Skip to content
Lint

[JAX] Fix get_seqlens_and_offsets() to accept vmapped seg ids and non vmapped seg offsets #15199

Sign in to view logs

[JAX] Fix get_seqlens_and_offsets() to accept vmapped seg ids and non vmapped seg offsets

[JAX] Fix get_seqlens_and_offsets() to accept vmapped seg ids and non vmapped seg offsets #15199

Triggered via pull request February 27, 2026 19:21
@KshitijLakhaniKshitijLakhani
synchronize #2692
KshitijLakhani:klakhani/fix/vmap-get-seg-ids-pos
Status Success
Total duration 2m 43s
Artifacts –

lint.yml

on: pull_request
PyTorch C++
33s
PyTorch C++
PyTorch Python
2m 25s
PyTorch Python
JAX C++
34s
JAX C++
JAX Python
31s
JAX Python
Fit to window
Zoom out
Zoom in