xla
Use f32 scratch for output so we only need to transfer output with desired dtype back to HBM.
#8924
Merged
Merged
Use f32 scratch for output so we only need to transfer output with desired dtype back to HBM. #8924
vanbasten23
merged
4 commits into
master
from
xiowei/migrate_kernel_change
Use f32 scratch for output so we only need to transfer output with de…
4e100747
not to use dynamic grid
e2cebd5b
linter
6a53e0a9
bythew3i
approved these changes
on 2025-04-02
change torch_xla wrapper
9383cbc5
yaochengji
approved these changes
on 2025-04-02
Login to write a write a comment.
Login via GitHub