Add torch 2.2.1 support #552

Job	Run time
build-image-matrix	11s
composer-0-20-0 / configure-build-push	34s
torch-2-0-1-cu118 / configure-build-push	1m 31s
torch-2-1-2-cu121 / configure-build-push	1m 23s
torch-nightly-2-3-0-20240110-cu121-python3-11 / configure-build-push	1m 36s
torch-2-2-1-cu121 / configure-build-push	1h 9m 19s
composer-0-20-0-cpu / configure-build-push	25s
torch-2-0-1-cpu / configure-build-push	23s
torch-2-1-2-cpu / configure-build-push	31s
torch-2-2-1-cpu / configure-build-push	3m 10s
torch-2-1-2-cu121-aws / configure-build-push	1m 5s
torch-nightly-2-3-0-20240110-cu121-python3-11-aws / configure-build-push	52s
torch-2-0-1-cu118-aws / configure-build-push	27s
torch-2-2-1-cu121-aws / configure-build-push	1h 8m 27s
	2h 29m 54s

Provide feedback