[Scheduler] Add the fused task name to auto-scheduled kernels

hidet-org / hidet

An open-source efficient deep learning framework/compiler, written in python.

https://hidet.org

Apache License 2.0

634 stars 50 forks source link

[Scheduler] Add the fused task name to auto-scheduled kernels #418

Closed yaoyaoding closed 6 months ago

yaoyaoding commented 6 months ago

This PR makes the generated kernel name contained more information. Previously, the name might look like hidet_compute_c, now it might look like hidet_fused_softmax_rearrange_c.