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

[inductor] don't append None to choices (#161672)

Browse Source 
Summary: don't append None as a choice to choices in autotune

Test Plan: See internal Diff

Differential Revision: D81188644

Pull Request resolved: https://github.com/pytorch/pytorch/pull/161672
Approved by: https://github.com/angelayi
This commit is contained in:
Shangdi Yu
2025-08-28 18:48:50 +00:00
committed by PyTorch MergeBot
parent 049c08eda8
commit 30ab87c884
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
torch/_inductor/select_algorithm.py
View File

@ -1453,7 +1453,9 @@ class TritonTemplate(KernelTemplate):
"""
try:
choices.append(self.generate(generate_with_caching=True, **kwargs))
choice = self.generate(generate_with_caching=True, **kwargs)
if choice is not None:
choices.append(choice)
return None
except NotImplementedError as e:
log.info(