[Stream] Enable batch affinity queries in SpecializeEncoding pass. #17604

Job	Run time
setup / setup	9s
Build Packages / Linux Release (x86_64)	7m 59s
Test AMD MI250 / test_mi250	3m 19s
Unit Test / Linux (x86_64)	5m 33s
Test AMD MI300 / test_mi300	13m 9s
Regression Test / test_regression_suite :: cpu_llvm_task	12m 14s
Regression Test / test_regression_suite :: amdgpu_rocm_mi250_gfx90a	16m 12s
Test AMD W7900 / test_w7900	2m 23s
Regression Test / test_regression_suite :: amdgpu_rocm_mi300_gfx942	11m 42s
Test Android / android_arm64	7m 3s
Test ONNX / test_onnx_models :: cpu_llvm_task	2m 25s
Regression Test / test_regression_suite :: amdgpu_rocm_mi308_gfx942	14m 56s
Test ONNX / test_onnx_models :: amdgpu_rocm_rdna3	1m 41s
Test ONNX / test_onnx_ops :: cpu_llvm_sync	4m 23s
Test ONNX / test_onnx_models :: amdgpu_vulkan	1m 26s
Test ONNX / test_onnx_ops :: amdgpu_rocm_rdna3	9m 8s
Test PJRT plugin / Build and test (ubuntu-24.04, cpu)	3m 7s
Test ONNX / test_onnx_ops :: amdgpu_vulkan	1m 50s
Test RISC-V 64 / riscv64	9m 44s
Test Sharktank / test_sharktank_models :: cpu_task	37s
Test TensorFlow / Linux (x86_64)	1m 3s
Test Sharktank / test_sharktank_models :: hip_task	39s
pkgci_summary / summary	1s
	2h 10m 43s

Provide feedback