Skip to content

[upstream #3549] [XPU] Enable functionalize_rng_ops on XPU (philox_rand prim + RNG decompositions)#21

Draft
Stonepia wants to merge 1 commit into
mainfrom
agent/issue-384
Draft

[upstream #3549] [XPU] Enable functionalize_rng_ops on XPU (philox_rand prim + RNG decompositions)#21
Stonepia wants to merge 1 commit into
mainfrom
agent/issue-384

Conversation

@Stonepia

Copy link
Copy Markdown
Collaborator

[upstream pytorch#3549] [XPU] Enable functionalize_rng_ops on XPU (philox_rand prim + RNG decompositions)

Fixes https://github.com/intel-sandbox/torch-xpu-ops-exp/issues/384

Root Cause: The functionalize_rng_ops path hardcodes CUDA-only device checks via throw_on_non_cuda in three locations (decompositions_for_rng.py lines 34, 55; rng_prims.py line 118). XPU supports Philox RNG but is rejected by these gates. Additionally, philox_rand_offset likely queries CUDA-specific device properties.


Diff stat:

torch/_decomp/decompositions_for_rng.py | 11 ++++++++--
 torch/_prims/rng_prims.py               | 38 +++++++++++++++++++++++++++------
 torch/_prims_common/__init__.py         | 20 ++++++++++++++++-
 3 files changed, 59 insertions(+), 10 deletions(-)

[upstream pytorch#3549] [XPU] Enable functionalize_rng_ops on XPU (philox_rand prim + RNG decompositions)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant