projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: e3b1758) | patch
Fix bug in tensor feed which caused crash due to wrong tensor type (#18552)
authorGu, Jinghui <jinghui.gu@intel.com>
Fri, 29 Mar 2019 21:06:09 +0000 (14:06 -0700)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Fri, 29 Mar 2019 21:12:36 +0000 (14:12 -0700)
commit84f020fe098bfaad308bd820869fafa44d3d4fff
treed17104d2ec28b09967d154adf91b09a49f8d3a16tree | snapshot
parente3b1758f19018d02ff0a509301cbad32cfe90aeacommit | diff
Fix bug in tensor feed which caused crash due to wrong tensor type (#18552)

Summary:
In blob feeder for ideep device, the wrong device option is given and led to a crash issue.
This patch aims to correct the device option to fix this bug.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/18552

Differential Revision: D14679838

Pulled By: yinghai

fbshipit-source-id: bde11e6a6fe44822166881dcb7c9bd0b34b4ecf3
caffe2/python/pybind_state_ideep.cc diff | blob | history
Domain: Machine Learning / ML Framework;