Support multinomial_from_uniform dispatch - Githubissues

apache / tvm

Open deep learning compiler stack for cpu, gpu and specialized accelerators

https://tvm.apache.org/

Apache License 2.0

11.67k stars 3.45k forks source link

Support multinomial_from_uniform dispatch #17010

Closed Hzfengsy closed 4 months ago

Hzfengsy commented 4 months ago

This PR aims to support backend dispatching for multinomial_from_uniform, which includes:

Relax Op multinomial_from_uniform
TIR gpu kernel for multinomial_from_uniform
dispatching pass
TVMScript parser support for pure-python If branch

cc @tqchen

tqchen commented 4 months ago

@tvm-bot rerun

tqchen commented 4 months ago

@tvm-bot rerun

tqchen commented 4 months ago

@tvm-bot rerun

yongwww commented 4 months ago

@tvm-bot rerun

tqchen commented 4 months ago

@tvm-bot rerun

tqchen commented 4 months ago

@tvm-bot rerun

tqchen commented 4 months ago

@tvm-bot rerun