[ir] Use IsCompute/IsFragment/IsVertex in more places #4720

Job	Run time
Go Build and Test	45s
CMake-windows-latest-Debug-msvc	13m 58s
CMake-windows-latest-Release-msvc	16m 39s
CMake-ubuntu-latest-Debug-gcc	21m 40s
CMake-ubuntu-latest-Release-gcc	20m 39s
CMake-macos-latest-Debug-clang	18m 2s
CMake-macos-latest-Release-clang	37m 5s
CMake-macos-13-Debug-clang	52m 44s
CMake-macos-13-Release-clang	1h 4m 34s
	4h 6m 6s

Provide feedback