Repository navigation
Add jagged_index_select_2d_forward XPU operator - #92
Merged
dvrogozh merged 10 commits intoAug 28, 2026
Merged
Conversation
flezaalv
changed the title
aagalleg
force-pushed
the
flezaalv/refactor_jagged_index_select_2D
branch
from
July 22, 2026 20:59
c641b23 to
8114ad5
Compare
Contributor
Author
Added upstream FBGEMM test coverage for |
flezaalv
force-pushed
the
flezaalv/refactor_jagged_index_select_2D
branch
from
August 27, 2026 21:34
b936395 to
f134ba6
Compare
Contributor
Author
|
Latest changes: CI (
Documentation (
Validation
|
flezaalv
marked this pull request as ready for review
August 27, 2026 23:53
flezaalv
requested review from
dvrogozh,
eromomon and
luis-real
as code owners
August 27, 2026 23:53
dvrogozh
requested changes
Aug 28, 2026
Contributor
|
Also, please, rebase as there is a conflict in the patch file due to recently merged PRs. |
Signed-off-by: Felipe Leza Alvarez <felipe.leza.alvarez@intel.com>
Signed-off-by: Felipe Leza Alvarez <felipe.leza.alvarez@intel.com>
Update device selection order in the test patch to check for XPU
availability first, then fall back to CPU or CUDA. This ensures XPU
tests run on XPU hardware when available, rather than defaulting to
CPU even when XPU is present.
Changes the device logic from:
"cpu" if use_cpu else ("xpu" if xpu_available else "cuda")
to:
"xpu" if xpu_available else ("cpu" if use_cpu else "cuda")
…2d tests Signed-off-by: Felipe Leza Alvarez <felipe.leza.alvarez@intel.com>
Signed-off-by: Felipe Leza Alvarez <felipe.leza.alvarez@intel.com>
Signed-off-by: Felipe Leza Alvarez <felipe.leza.alvarez@intel.com>
Signed-off-by: Felipe Leza Alvarez <felipe.leza.alvarez@intel.com>
Signed-off-by: Felipe Leza Alvarez <felipe.leza.alvarez@intel.com>
…order with CUDA Signed-off-by: Felipe Leza Alvarez <felipe.leza.alvarez@intel.com>
flezaalv
force-pushed
the
flezaalv/refactor_jagged_index_select_2D
branch
from
August 28, 2026 18:25
8467cd1 to
a31766f
Compare
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
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.
This commit introduces the
jagged_index_select_2d_forwardoperator for Intel XPU devices, enabling 2D index selection on jagged tensors with offset-based indexing.Changes
New SYCL Kernel Implementation (
jagged_index_select_2d.cpp/h)JaggedIndexSelect2dKernelfunctor for SYCL/XPU executionOperator Registration (
ops_registry.cpp)fbgemmnamespaceBuild Integration (
CMakeLists.txt)cc: @aagalleg, @manuelhsantana