fix kMaxInputCount check in GroupedMatmulFunctor #77902

Job	Run time
Collect information about PR and source	14s
Mirror third party dependencies	1m 2s
License and format	1m 29s
Find build cache	13s
Static analysis with clang on diff	1h 1m 33s
Build OneFlow (cu116)	22m 32s
Build OneFlow (cpu)	13m 40s
Build OneFlow (cpu-asan-ubsan)	23s
Build OneFlow (cpu-tsan)	27s
Build OneFlow (llvm15)	22m 52s
Find test cache	13s
Find test cache (distributed)	0s
Test suite (cuda-module)	38m 34s
Test suite (cuda-misc)	38m 51s
Test suite (cuda-speed-test)	23m 48s
Test suite (cpu-module)	23m 12s
Test suite (cpu-misc)	20m 15s
Distributed test suite	0s
	4h 29m 18s

Provide feedback