Support unfixed kv heads number (#1416) #6178

Job	Run time
cpp-beam_search_causal_lm-notus-7b-v1	31m 4s
cpp-beam_search_causal_lm-Phi-2	16m 36s
cpp-beam_search_causal_lm-Qwen1_5-7B-Chat	33m 4s
cpp-beam_search_causal_lm-ubuntu (./build/samples/cpp/beam_search_causal_lm/beam_search_causal_lm)	19m 33s
cpp-chat_sample-ubuntu	14m 52s
cpp-continuous-batching-macos	22m 20s
cpp-beam_search_causal_lm-ubuntu (python ./samples/python/beam_search_causal_lm/beam_search_causa...	19m 26s
cpp-continuous-batching-ubuntu	13m 12s
cpp-continuous-batching-windows	27m 32s
cpp-greedy_causal_lm-Qwen-7B-Chat	10m 38s
cpp-greedy_causal_lm-redpajama-3b-chat	11m 24s
cpp-greedy_causal_lm-windows	27m 15s
cpp-multinomial-greedy_causal_lm-ubuntu	18m 31s
cpp-Phi-1_5	9m 7s
cpp-prompt_lookup_decoding_lm-ubuntu	9m 39s
cpp-speculative_decoding_lm-ubuntu	12m 5s
visual_language_chat_sample-ubuntu-internvl2	23m 26s
visual_language_chat_sample-ubuntu-minicpm_v2_6	7m 52s
visual_language_chat_sample-ubuntu-llava_1_5 / visual_language_chat_sample-ubuntu-llava	14m 52s
visual_language_chat_sample-ubuntu-llava_next / visual_language_chat_sample-ubuntu-llava	18m 36s
ci/gha_overall_status_causal_lm	1s
	6h 1m 5s

Provide feedback