JackCaoG
JackCaoG
Thanks @ngam for the heads up, I will try to test after we update our tf pins.
let's rebase again, I will try to review it today or tmr.
you need to update this pr a bit to account for new tf update, should not be too much code changes through
> > you need to update this pr a bit to account for new tf update, should not be too much code changes through > > Do we have a...
Thanks, I will take a look later today
Seems like test strict out crashed. @yeounoh were you able to run a simple pytorch/xla program locally?
Seesm irrelevant, let me just restart the gpu ci
I will take another pass and try to merge it.
https://github.com/pytorch/pytorch/pull/82970#pullrequestreview-1065922141 fixed the device issue.
Issue coming from trying to add f64 and f32 ``` 2 root error(s) found. (0) INTERNAL: during context [Unknown]: Seen floating point types of different precisions in %add.122 = f64[1,16]{1,0}...