[TPU][Pallas]Fix example/cross_entropy.py on Pallas TPU#2002

Open

yarongmu-google wants to merge 11 commits intopytorch:mainfrom

yarongmu-google:fix-pallas-dtype-mapping

Commits on Apr 3, 2026

[Pallas] Add test for Pallas OOB slice when reduction_loops doesn't divide dim (#1937)

norx1991
authored and
yarongmu-google
committed
Merge branch 'pytorch:main' into main
yarongmu-google
authored

Commits on Apr 9, 2026

[Pallas] Add test for Pallas OOB slice when reduction_loops doesn't divide dim (#1937)

norx1991
authored and
yarongmu-google
committed
Merge branch 'pytorch:main' into main
yarongmu-google
authored
fix(pallas): add mapping for 64-bit dtypes to 32-bit to avoid Pallas errors and fix zero division in block size calculation
yarongmu-google
committed

Commits on Apr 13, 2026