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

functionalization <> LTC integration (take 3) (#80251)

Browse Source 
new PR for https://github.com/pytorch/pytorch/pull/75527.

It looks like there's a bug in the windows CI scripts that was causing
flaky failures, that disappear when I create a new PR. example failure:
https://github.com/pytorch/pytorch/runs/6999272635?check_suite_focus=true
Pull Request resolved: https://github.com/pytorch/pytorch/pull/80251
Approved by: https://github.com/wconstab
This commit is contained in:
Brian Hirsh
2022-06-26 07:42:42 -07:00
committed by PyTorch MergeBot
parent 33761c80d2
commit c2d395cf8e
27 changed files with 1059 additions and 294 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
torchgen/gen_backend_stubs.py
View File

@ -265,7 +265,10 @@ def error_on_missing_kernels(
native_f
)
kernel_defn_regex = rf"{class_name}::([\w\d]*)\([^\)]*\)\s*{{"
# This just looks for lines containing "foo(", and assumes that the kernel foo has been implemented.
# It might cause false negatives (we won't catch all cases), but that's ok - if we catch a missing kernel
# here, then we get a nicer error message. If we miss it, you get a linker error.
kernel_defn_regex = rf"{class_name}::\s*([\w\d]*)\("
actual_backend_kernel_name_counts = Counter(
re.findall(kernel_defn_regex, backend_defns)
)