fix test failure of speculative_generation on xpu (#42052)
* fix test failure of speculative_generation on xpu
Signed-off-by: Wang, Yi A <yi.a.wang@intel.com>
* code refine
Signed-off-by: Wang, Yi A <yi.a.wang@intel.com>
* address review comment
Signed-off-by: Wang, Yi A <yi.a.wang@intel.com>
---------
Signed-off-by: Wang, Yi A <yi.a.wang@intel.com>