tvm icon indicating copy to clipboard operation
tvm copied to clipboard

Published 20 hours ago • verified publisher icon apache

Support multinomial_from_uniform dispatch

Open Hzfengsy opened this issue 1 year ago • 5 comments

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

Hzfengsy avatar May 20 '24 05:05 Hzfengsy

@tvm-bot rerun

tqchen avatar May 21 '24 12:05 tqchen

@tvm-bot rerun

tqchen avatar May 22 '24 18:05 tqchen

@tvm-bot rerun

tqchen avatar May 23 '24 11:05 tqchen

@tvm-bot rerun

yongwww avatar May 23 '24 19:05 yongwww

@tvm-bot rerun

tqchen avatar May 23 '24 23:05 tqchen

@tvm-bot rerun

tqchen avatar May 23 '24 23:05 tqchen

@tvm-bot rerun

tqchen avatar May 24 '24 03:05 tqchen