Migrate GenAI kv cache kernel arguments to use `PTA_B` (#4899) #18745

Job	Run time
build-bazel (x86, linux.12xlarge, clang)	9m 38s
build-macos (macos-13, shared)	26m 14s
build-linux (arm, linux.arm64.m7g.4xlarge, static, clang, 16.0.6)	33m 15s
build-macos (macos-13, static)	28m 9s
build-windows (windows-2022, static)	12m 22s
build-linux (x86, linux.12xlarge, static, gcc, 9.5.0)	25m 55s
build-bazel (x86, linux.12xlarge, gcc)	10m 2s
build-linux (x86, linux.12xlarge, static, clang, 16.0.6)	20m 4s
build-linux (x86, linux.12xlarge, shared, gcc, 14.1.0)	26m 23s
build-linux (arm, linux.arm64.m7g.4xlarge, shared, clang, 16.0.6)	33m 6s
build-linux (x86, linux.12xlarge, shared, clang, 16.0.6)	19m 56s
build-linux (x86, linux.12xlarge, shared, gcc, 9.5.0)	25m 40s
build-windows (windows-2022, shared)	12m 45s
build-linux (x86, linux.12xlarge, static, gcc, 14.1.0)	26m 25s
build-linux (arm, linux.arm64.m7g.4xlarge, shared, gcc, 14.1.0)	1h 23m 11s
build-linux (arm, linux.arm64.m7g.4xlarge, static, gcc, 14.1.0)	1h 25m 58s
	7h 59m 3s

Provide feedback