projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 3cba115) | patch
Fix the tensor deserialization problem of jit script module on CUDA (#16279)
authorLu Fang <lufang@fb.com>
Thu, 24 Jan 2019 05:32:57 +0000 (21:32 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Thu, 24 Jan 2019 05:35:35 +0000 (21:35 -0800)
commit8ab4d348f464fe9bf28bf6cbf518f97a65efae1b
tree483429544df8e49f88e0f9023bad9e6c656aaa8btree | snapshot
parent3cba115abb750b74ecc246f76e0592ca3ca751a9commit | diff
Fix the tensor deserialization problem of jit script module on CUDA (#16279)

Summary:
Now we create a temporary tensor for the whole record.

Fix https://github.com/pytorch/pytorch/issues/15271
Pull Request resolved: https://github.com/pytorch/pytorch/pull/16279

Reviewed By: BIT-silence

Differential Revision: D13791442

Pulled By: houseroad

fbshipit-source-id: 6f52ca09627fb684f74121357cc42e4adadec36a
test/test_jit.py diff | blob | history
torch/csrc/jit/import.cpp diff | blob | history
Domain: Machine Learning / ML Framework;