port test_sparse and test_sparse_csr for Intel GPU #2252
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
wincent8
force-pushed
the
wliao2/enable_sparse
branch
from
87aecd5 to
143ea51
Compare
According to https://github.com/intel/torch-xpu-ops/actions/runs/18909341402 disable_build disable_e2e disable_distributed
To solve #2207 This PR adds FP8 data types support for `torch.cat` and `torch.where` on XPU backend. --------- Co-authored-by: Cui, Yifeng <[email protected]> Co-authored-by: Copilot <[email protected]>
To solve #2207 Extends support for float8 data types across various XPU tensor indexing and transformation kernels, ensuring these operations are compatible with the new types. It also adds a regression test for flipping float8 tensors and removes the skip for float8 indexing tests. **Float8 type support:** * Updated dispatch macros in `XPUScalar.cpp` and `Indexing.cpp` to include `AT_FLOAT8_TYPES`, enabling float8 support in scalar extraction, indexing, index_put, and deterministic index_put kernels. * Modified `flip_kernel` in `TensorTransformationsKernels.cpp` to support float8 and barebones unsigned types, updating the dispatch mechanism accordingly. * Included the new dispatch header `Dispatch_v2.h` for the updated dispatch macros. **Testing improvements:** * Added a regression test for flipping float8 tensors in `test_index_and_index_put.py` to verify correctness of the operation on XPU. * Removed the skip for float8 tests in `test_indexing_xpu.py`, re-enabling these tests now that support is implemented. --------- Co-authored-by: Cui, Yifeng <[email protected]>
…-xpu-ops into wliao2/enable_sparse
This was referenced Nov 5, 2025
Open
Open
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
disable_distributed
disable_e2e
disable_build