Repository navigation
CUDA: support strided ABS for F16 and F32 - #28477
mornhussakuyo-hub wants to merge 2 commits into
Conversation
Assisted-by: Codex
|
Hi @ggerganov , just a friendly ping on this PR. 🙂 I noticed that the CI workflows are currently showing The change has also been tested locally on CUDA, and I'd really appreciate a review when you have time. Thanks! |
|
Thank you for your efforts. Regrettably this PR is implementing the wrong thing. Instead of a special case for |
|
Looking further, there seems to already be a PR: #28821 . |
Assisted-by: Codex
Overview
CUDA ABS previously only supported fully contiguous tensors.
This PR adds support for tensors with contiguous rows and arbitrary outer strides, while preserving the existing fast path for fully contiguous tensors. Source and destination strides are handled independently, allowing ABS to work correctly with offset views, permuted outer dimensions, and in-place operations.
This contributes to the CUDA operator support work tracked in #14909 by filling the missing strided-tensor support for ABS.
Supported types:
Tests were added for padded and offset views, permuted outer dimensions, and both regular and in-place execution.
Additional information
The implementation was validated locally on an NVIDIA RTX 5060 Ti. The expanded ABS tests pass for all covered F16/F32 cases, including strided and in-place layouts.
Requirements