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

[functorch] quick fix

Browse Source
This commit is contained in:
Richard Zou
2021-06-14 08:40:18 -07:00
committed by Jon Janzen
parent 4a045e9659
commit cfa9d98499
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
functorch/codegen/codegen_outofplacebatching.py
View File

@ -163,7 +163,7 @@ def get_signatures(path='build/aten/src/ATen/RegistrationDeclarations.h', includ
continue
if 'std::array' in line:
continue
m = re.match(r'(.*) \w+\((.*)\); // {"schema": "aten::(.*)\(.*', line)
m = re.match(r'(.*) \w+\((.*)\); // {"schema": "aten::(\w+\.?\w*)\(.*', line)
if m is None:
continue
return_t = m.group(1)