transformers

dcf0c7c8 - fbgemm_fp8:Keep the current device aligned with the input tensor (#46403)

fbgemm_fp8:Keep the current device aligned with the input tensor (#46403) * fbgemm_fp8:Keep the current device aligned with the input tensor Signed-off-by: kaixuanliu <kaixuan.liu@intel.com> * update comment Signed-off-by: kaixuanliu <kaixuan.liu@intel.com> * move `on_device` func to common part Signed-off-by: kaixuanliu <kaixuan.liu@intel.com> * update Signed-off-by: kaixuanliu <kaixuan.liu@intel.com> * update code Signed-off-by: kaixuanliu <kaixuan.liu@intel.com> --------- Signed-off-by: kaixuanliu <kaixuan.liu@intel.com>