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] address comments from https://github.com/pytorch/pytorch/pull/163803 (#163901)

Browse Source 
Summary: address comments from https://github.com/pytorch/pytorch/pull/163803

Differential Revision: D83291637

Pull Request resolved: https://github.com/pytorch/pytorch/pull/163901
Approved by: https://github.com/desertfire
This commit is contained in:
Huamin Li
2025-09-26 17:18:44 +00:00
committed by PyTorch MergeBot
parent 5880996b4c
commit 9534c59311
1 changed files with 2 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
torch/_inductor/decomposition.py
View File

@ -885,12 +885,9 @@ def select_decomp_table() -> dict[Any, Callable[..., Any]]:
"""decomps can change based on config"""
if config.fallback_random:
return decompositions
if config.fallback_embedding_bag_byte_unpack is True:
if config.fallback_embedding_bag_byte_unpack:
# remove q_embedding_bag_byte_unpack_decomp from decompositions
for k in decompositions.keys():
if k == torch.ops.quantized.embedding_bag_byte_unpack.default:
del decompositions[k]
break
decompositions.pop(torch.ops.quantized.embedding_bag_byte_unpack.default, None)
return decompositions
return fast_random_decomps()