Skip to content
This repository has been archived by the owner on Feb 1, 2020. It is now read-only.

Commit

Permalink
fix windows compilation error (#115)
Browse files Browse the repository at this point in the history
  • Loading branch information
eric-haibin-lin authored and piiswrong committed May 18, 2017
1 parent 7c603eb commit 31920d7
Showing 1 changed file with 3 additions and 4 deletions.
7 changes: 3 additions & 4 deletions src/pass/infer_shape_type.cc
Original file line number Diff line number Diff line change
Expand Up @@ -260,14 +260,13 @@ NNVM_REGISTER_PASS(InferStorageType)
.describe("Infer the storage type of each node entries.")
.set_body([](Graph ret) {
// for storage type, the backward attr is not necessarily the same as it's correspondence
const int none = -1;
const int kDefaultStorage = 0;
return InferAttr<int>(
std::move(ret), none,
std::move(ret), -1,
"FInferStorageType", "storage_type_inputs", "storage_type_attr_key",
"storage_type", "storage_type_num_unknown_nodes",
[](const int t) { return t == none; },
DefaultType<kDefaultStorage, none>, false);
[](const int t) { return t == -1; },
DefaultType<kDefaultStorage, -1>, false);
})
.set_change_graph(false)
.provide_graph_attr("storage_type");
Expand Down

0 comments on commit 31920d7

Please sign in to comment.