pytorch/android/pytorch_android/src
Jorge Pineda 93e6fc54fa [PyTorch] Remove device transfers from JNI (#105583)
Summary:
If a model was exported for Vulkan backend without (automatic or manual) device transfers, then the export is incorrect, and the JNI need not correct that.
(If this assumption is incorrect, please give feedback.)

Undo the changes from
- D23763771: automatic device transfers in JNI
- D39519168: `"requires_backend_transfers"` logic in JNI

Test Plan: Verify CUNET+ hybrid model from D47488843 works.

Reviewed By: SS-JIA

Differential Revision: D47527244

Pull Request resolved: https://github.com/pytorch/pytorch/pull/105583
Approved by: https://github.com/SS-JIA
2023-07-20 00:26:21 +00:00
..
androidTest
main [PyTorch] Remove device transfers from JNI (#105583) 2023-07-20 00:26:21 +00:00