fix kMaxInputCount check in GroupedMatmulFunctor #77901

Job	Run time
License and format	1m 26s
Find build cache	13s
Collect information about PR and source	15s
Mirror third party dependencies	1m 12s
Static analysis with clang on diff	37m 19s
Build OneFlow (cu116)	25m 13s
Build OneFlow (cpu)	14m 2s
Build OneFlow (cpu-asan-ubsan)	28s
Build OneFlow (cpu-tsan)	36s
Build OneFlow (llvm15)	19m 33s
Find test cache	11s
Find test cache (distributed)	0s
Test suite (cuda-module)	11m 29s
Test suite (cuda-misc)	11m 25s
Test suite (cuda-speed-test)	11m 29s
Test suite (cpu-module)	11m 26s
Test suite (cpu-misc)	11m 26s
Distributed test suite	1s
	2h 37m 44s

Provide feedback