xla icon indicating copy to clipboard operation
xla copied to clipboard

Published 20 hours ago • verified publisher icon pytorch

Tracking issue: PyTorch precision upcast issue.

Open ysiraichi opened this issue 1 year ago • 0 comments

This issue tracks the development of the output upcasting issue present in PyTorch/XLA.

  • [ ] Main issue: #6403
  • [x] Temporary fix for the benchmarking scripts: #6389
  • [x] PyTorch nn.Module conversion fix: https://github.com/pytorch/pytorch/pull/117167

Once pytorch#117167 lands, we should revert pytorch/xla#6389, and stop setting XLA_USE_{FP16,BF16} in the benchmarking script.

cc @miladm @JackCaoG @golechwierowicz

ysiraichi avatar Jan 29 '24 19:01 ysiraichi