frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-21 05:34:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

Improve argument matching in invalidArguments

Browse Source
This commit is contained in:
Adam Paszke
2016-09-29 11:53:20 -07:00
parent c8a4734b97
commit a33c9bd774
1 changed files with 12 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
torch/csrc/utils.cpp
View File

@ -152,8 +152,18 @@ std::string THPUtils_formattedTupleDesc(
for (size_t i = 0; i < argument_types.size(); i++) {
bool is_matching = false;
if (i < expected_arguments.size()) {
is_matching = argument_types[i] == expected_arguments[i].type ||
(expected_arguments[i].is_nullable && argument_types[i] == "NoneType");
if (expected_arguments[i].type == "float") {
is_matching = argument_types[i] == "float" ||
argument_types[i] == "int" ||
argument_types[i] == "long";
} else if (expected_arguments[i].type == "int") {
is_matching = argument_types[i] == "int" ||
argument_types[i] == "long";
} else {
is_matching = expected_arguments[i].type == argument_types[i];
}
if (expected_arguments[i].is_nullable && argument_types[i] == "NoneType")
is_matching = true;
} else if (is_variadic) {
is_matching = argument_types[i] == expected_arguments.back().type;
}