saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix test_rng bisector test (#143662)

Browse source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/143662
Approved by: https://github.com/zou3519
This commit is contained in:
eellison 2025-01-17 17:18:53 +00:00 committed by PyTorch MergeBot
parent d02c396fbb
commit c338dda6be
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
torch/_inductor/inductor_prims.py
View file

@ -65,7 +65,7 @@ seeds = make_prim(
lookup_seed = make_prim(
# if inductor_lookup_seed changes, update partitioners.py
"inductor_lookup_seed(Tensor seeds, int index) -> Tensor",
lambda seeds, index: seeds[index],
lambda seeds, index: seeds[index].clone(),
doc="Extract a single seed from the result of inductor_seeds()",
)
# inductor_random() doesn't accept a dtype.