frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-11-01 13:34:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

Consistently check 'out' variants against specified dtype/layout/device parameters. (#6973)

Browse Source 
We were previously doing this in the most common cases, but not consistently.
This commit is contained in:
gchanan
2018-04-25 22:46:42 -04:00
committed by GitHub
parent c10da636b5
commit 3d907ef78e
3 changed files with 29 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
tools/autograd/gen_python_functions.py
View File

@ -73,10 +73,9 @@ PY_VARIABLE_OUT_CHECK_TYPE = CodeTemplate("""\
if (r.isNone(${out_idx})) {
${call_dispatch}
} else {
if (!r.isNone(${type_idx})) {
check_out_type_matches(r.tensor(${out_idx}), r.scalartype(${type_idx}), r.layout(${layout_idx}),
r.device(${device_idx}), r.isNone(${device_idx}));
}
check_out_type_matches(r.tensor(${out_idx}), r.scalartype(${type_idx}), r.isNone(${type_idx}),
r.layout(${layout_idx}), r.isNone(${layout_idx}),
r.device(${device_idx}), r.isNone(${device_idx}));
${call_dispatch_out}
}
""")