Skip to content
Permalink
Browse files

Fix pytorch build (#3441)

Summary:
See title

Documentation:
none
Pull Request resolved: #3441

Test Plan:
python setup.py test
CI

Differential Revision: D16939660

Pulled By: jackm321

fbshipit-source-id: d485d10701cb086333432182ba3872eaa930e0c9
  • Loading branch information...
jackm321 authored and facebook-github-bot committed Aug 21, 2019
1 parent 63e16c6 commit 4f27105ca113d628bafc4f1aec393281955408d1
Showing with 1 addition and 3 deletions.
  1. +0 −1 .circleci/build.sh
  2. +1 −2 torch_glow/src/PyTorchModelLoader.cpp
@@ -128,7 +128,6 @@ elif [[ "$CIRCLE_JOB" == "PYTORCH" ]]; then
source venv/bin/activate
git clone https://github.com/pytorch/pytorch.git --recursive
cd pytorch
git checkout 7f86fb8995d9990fe1eac5a18335523cb3da7f09
pip install -r requirements.txt
python setup.py install
cd ${GLOW_DIR}
@@ -526,8 +526,7 @@ PyTorchModelLoader::loadValue(const torch::jit::Value *value) {
RETURN_ERR_IF_NOT(value->isCompleteTensor(),
glow::strFormat("Value %s must have CompleteTensor type.",
value->debugNameBase().c_str()));
auto glowType =
ptTypeToGlowType(*value->type()->expect<at::TensorType>());
auto glowType = ptTypeToGlowType(*value->type()->expect<at::TensorType>());
glow::Placeholder *ph = F_.getParent()->createPlaceholder(
&glowType, "input", /*isTrainable*/ false);
RETURN_IF_ERR(addGlowNodeValue(value, ph->getOutput()));

0 comments on commit 4f27105

Please sign in to comment.
You can’t perform that action at this time.