fix(megatron): cast get_cp_local_num_tokens to torch.int for Megatron int32 token accumulator - #290
Open
ZiyiTsang wants to merge 1 commit into
Open
fix(megatron): cast get_cp_local_num_tokens to torch.int for Megatron int32 token accumulator#290ZiyiTsang wants to merge 1 commit into
ZiyiTsang wants to merge 1 commit into
