[Test] Fix build break #312

Job	Run time
unit-tests-gpu (gpu-runner, 3.10, gpt2gpu) / unit-tests-on-gpu	20m 27s
unit-tests-gpu (gpu-runner, 3.10, hfllama_7b_gpu) / unit-tests-on-gpu	15m 15s
unit-tests-gpu (gpu-runner, 3.10, phi2gpu) / unit-tests-on-gpu	14m 12s
unit-tests-gpu (gpu-runner, 3.11, gpt2gpu) / unit-tests-on-gpu	14m 39s
unit-tests-gpu (gpu-runner, 3.11, hfllama_7b_gpu) / unit-tests-on-gpu	15m 32s
unit-tests-gpu (gpu-runner, 3.12, gpt2gpu) / unit-tests-on-gpu	15m 33s
unit-tests-gpu (gpu-runner, 3.11, phi2gpu) / unit-tests-on-gpu	14m 24s
unit-tests-gpu (gpu-runner, 3.12, hfllama_7b_gpu) / unit-tests-on-gpu	16m 49s
unit-tests-gpu (gpu-runner, 3.12, phi2gpu) / unit-tests-on-gpu	14m 56s
unit-tests-gpu (gpu-runner, 3.8, gpt2gpu) / unit-tests-on-gpu	14m 39s
unit-tests-gpu (gpu-runner, 3.8, hfllama_7b_gpu) / unit-tests-on-gpu	15m 18s
unit-tests-gpu (gpu-runner, 3.8, phi2gpu) / unit-tests-on-gpu	20m 52s
unit-tests-gpu (gpu-runner, 3.9, gpt2gpu) / unit-tests-on-gpu	14m 40s
unit-tests-gpu (gpu-runner, 3.9, hfllama_7b_gpu) / unit-tests-on-gpu	15m 36s
unit-tests-gpu (gpu-runner, 3.9, phi2gpu) / unit-tests-on-gpu	14m 15s
	3h 57m 7s

Provide feedback