pytorch
6f13d689 - Add meta support for multinomial (#101324)

Commit
1 year ago
Add meta support for multinomial (#101324) # Summary Found this when trying to compile the text gen loop of nanogpt here: https://github.com/pytorch/benchmark/blob/b33289942ba326703518c819b32c902dabdee8c1/torchbenchmark/models/nanogpt_generate/model.py#L322 Pull Request resolved: https://github.com/pytorch/pytorch/pull/101324 Approved by: https://github.com/ngimel
Author
Committer
Parents
Loading