fix change introduced by PT#21852

[ailzhang]

pytorch/pytorch#21852

[dlibenzi]

Ouch ... keep biting us:

RuntimeError: Attempted to call `variable.set_data(tensor)`, but `variable` and `tensor` have different types of TensorImpl.

[dlibenzi]

Oh, OK ... one HUNK failed:

+ xla/scripts/apply_patches.sh
1 out of 1 hunk FAILED
Applying patch file: /tmp/pytorch/xla/torch_patches/21048.diff

Is Will's new PR in?

[dlibenzi]

This PR though is benign enough to go in regardless.

[ailzhang]

@dlibenzi Yea the first PR of @yf225 's fix (pytorch/pytorch#21865) went in last night. And the last one (pytorch/pytorch#21613) is approved, should be merged soon I think. We can safely remove the patch after that. (We've tested it locally yesterday, should be good).