Fix autotune to support Triton 2.1 (#4340)
* need to change autotune and config prunning codes for triton 2.1
* formatting
---------
Co-authored-by: Stephen Youn <styoun@microsoft.com>
Co-authored-by: Lev Kurilenko <113481193+lekurile@users.noreply.github.com>
