Searched defs:split_id (Results 1 – 5 of 5) sorted by relevance
61 uint64_t skel_id, split_id; in main() local
212 int64_t split_id = iter - group_devices.begin(); in GetSplitIdAndRank() local216 std::tuple<int64_t, int64_t> PromptFlashAttentionInfo::GetAttenionMaskAttrs(const int64_t split_id, in GetAttenionMaskAttrs()451 int64_t split_id = GetSplitIdAndRank(); in ReplaceNodeInputOrAttrs() local
937 int64_t split_id = iter - group_devices.begin(); in GetSplitIdAndRank() local943 std::tuple<int64_t, int64_t> FlashAttentionScoreInfo::GetAttentionMaskAttrs(const int64_t split_id, in GetAttentionMaskAttrs()979 int64_t split_id = split_info[kIndex2]; in ReplaceActualSeqLenForSplitSeqInTnd() local1031 int64_t split_id = split_info[kIndex2]; in ReplaceNodeInputOrAttrs() local1135 void FlashAttentionScoreInfo::GetFlashAttentionScoreOpNode(int64_t split_id, int64_t split_num, con… in GetFlashAttentionScoreOpNode()1220 int64_t split_id = split_info[kIndex2]; in ComputeReplaceGraphForLoadBalance() local
360 int64_t split_id = iter - group_devices.begin(); in GetSplitIdAndRank() local382 …tuple<int64_t, int64_t> FusedInferAttentionScoreInfo::GetAttentionMaskAttrs(const int64_t split_id, in GetAttentionMaskAttrs()561 int64_t split_id = GetSplitIdAndRank(); in ReplaceNodeInputOrAttrs() local
11896 uint64_t split_id; in print_debug() local