nit: This function is a bit confusing; better use two separate functions, one to check the `triton` and another for `mxfp4_forward_kernel_wrapper `. _Originally posted by @yiliu30 in https://github.com/intel/auto-round/pull/1515#discussion_r2938908723_