PR #4848 Multinomial op lowering

Multinomial op lowering #4848

cowanmeg merged 8 commits into pytorch:master from cowanmeg:multinomial

Start multinomial lowering

365eef9a

add files

f8790b63

wonjoo-wj marked this pull request as draft 2 years ago

wonjoo-wj requested a review from

wonjoo-wj 2 years ago

miladm commented on 2023-04-04

multinomial working

9eb24a81

Fix test, format

9604b62b

cowanmeg changed the title ~~[WIP] Multinomial op lowering~~ Multinomial op lowering 2 years ago

cowanmeg added lowering

cowanmeg marked this pull request as ready for review 2 years ago

wonjoo-wj commented on 2023-04-07

fix long compilation. lower directly to xla loop

fb24cc63

address comments

ce4bc24e

disable pytorch test and address comments

48877c93

lint

5fa7c896

wonjoo-wj approved these changes on 2023-04-10

cowanmeg merged b437a331 into master 2 years ago

Reviewers

wonjoo-wj

miladm

Assignees

No one assigned

Labels

lowering

Milestone

No milestone