[Unity][Solved] how to bind variable to thread environment in relax?

Development unity
#1

wudi
wudi960×735 41.9 KB
dilun
dilun1011×307 13.2 KB

Relax script compile into cuda got error
#2

What should i do to compile GPU target code?

#3

Would be great to learn how to optimize for gpu at 6.1. Part 1 — Machine Learing Compilation 0.0.1 documentation.

1 Like
#4

Thank you very much! I it solve my problem

#5

does relax have auto-schedule mechanism now? when many kind of ops are imported from relax front(such as torch FX)，should I bind variable to thread manully with cuda target?

#6

Meta-schedule is fully supported by relax for static shapes. Dynamic shape tuning is not ready now