PR #35016 Only cast `cu_seqlens` when tracing

Only cast `cu_seqlens` when tracing #35016

ArthurZucker merged 2 commits into main from fix-cu_seqlens

Only cast `cu_seqlens` when tracing

5ad87af4

Formatting

fb8eff7e

xenova requested a review from

zucchini-nlp 1 year ago

zucchini-nlp approved these changes on 2024-11-29

zucchini-nlp requested a review from

ArthurZucker 1 year ago

ruidazeng approved these changes on 2024-11-29

ArthurZucker approved these changes on 2024-12-02

ArthurZucker merged 3480cbb9 into main 1 year ago

ArthurZucker deleted the fix-cu_seqlens branch 1 year ago

Reviewers

ArthurZucker

zucchini-nlp

ruidazeng

Assignees

No one assigned

Labels

None yet

Milestone

No milestone