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

Reland 2 Many symintifications (#87604) (#87980)

Browse Source 
Fixes #ISSUE_NUMBER

Pull Request resolved: https://github.com/pytorch/pytorch/pull/87980
Approved by: https://github.com/ezyang
This commit is contained in:
albanD
2022-10-28 13:40:11 +00:00
committed by PyTorch MergeBot
parent ce3e0e9856
commit 8a9aca7b8d
29 changed files with 197 additions and 154 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tools/jit/gen_unboxing.py
View File

@ -116,7 +116,9 @@ class ComputeCodegenUnboxedKernels:
# from wrapping/unwrapping TensorOptios.
# However, we would look to include default args for schema parsing.
# Default args only show up in the nonfaithful C++ API,
arg_default = cpp.default_expr(arg.argument.default, arg.argument.type)
arg_default = cpp.default_expr(
arg.argument.default, arg.argument.type, symint=False
)
if arg_default.startswith("{"):
arg_cpp = f"c10::IntArrayRef({arg_default})"
else: