xla
Add helper functions getDefaultXLAGenerator and createXLAGenerator to XLA random number generator
#9682
Merged
Merged
Add helper functions getDefaultXLAGenerator and createXLAGenerator to XLA random number generator #9682
iwknow
merged
20 commits into
pytorch:master
from
iwknow:generator
Add xla random generator.
07ca2391
format cpp files
1d67a02d
Merge branch 'pytorch:master' into generator
2fbef1a5
Merge branch 'pytorch:master' into generator
d60c447c
Merge branch 'pytorch:master' into generator
6d5d2d2d
Merge branch 'pytorch:master' into generator
fa490994
Add helper functions `getDefaultXLAGenerator` and `createXLAGenerator…
79ff99bc
implement `XLAHooks` and register it to PyTorch when loaded.
79d4b423
format
914d7089
Revert "implement `XLAHooks` and register it to PyTorch when loaded."
1224531f
Add missing include
80b20783
Merge branch 'pytorch:master' into generator
60b408d4
improve error reporting and function naming.
c65ce8f1
format
e7453366
Add unit tests for `GetDefaultXLAGenerator` and `CreateXLAGenerator`
e89e659c
Improve InitXLAGenVector function and the unit tests accordingly.
4cc594fb
Address feedbacks.
c93e10db
ysiraichi
commented
on 2025-11-06
update `InitXLAGenVector` and move setting of `PJRT_DEVICE` and `CPU_…
639151f7
address feedbacks.
1a681aaa
ysiraichi
approved these changes
on 2025-11-13
format.
44947588
Login to write a write a comment.
Login via GitHub