use the correct available memory API for XPU #4993

Job	Run time
run-tests (latest, test_prod)	2m 13s
run-tests (latest, test_core)	3m 15s
run-tests (latest, test_cli)	2m 40s
run-tests (latest, test_big_modeling)	2m 28s
run-tests (latest, test_deepspeed)	2m 29s
run-tests (latest, test_fsdp)	3m 19s
run-tests (latest, test_example_differences)	2m 46s
run-tests (latest, test_checkpoint_step)	4m 13s
run-tests (latest, test_checkpoint_epoch)	3m 34s
run-tests (latest, test_rest)	9m 10s
run-tests (minimum, test_prod)	2m 40s
run-tests (minimum, test_core)	3m 47s
run-tests (minimum, test_cli)	3m 19s
run-tests (minimum, test_big_modeling)	3m 20s
run-tests (minimum, test_deepspeed)	3m 10s
run-tests (minimum, test_fsdp)	3m 16s
run-tests (minimum, test_example_differences)	8m 59s
run-tests (minimum, test_checkpoint_step)	10m 38s
run-tests (minimum, test_checkpoint_epoch)	4m 40s
run-tests (minimum, test_rest)	9m 50s
	1h 29m 46s

Provide feedback