padded_max_slots was computed from max_tokens*top_k (3072) but total_padded_slots in run() is num_experts*max_chunks*128 (6144). The buffer was too small, causing index out of bounds.
padded_max_slots was computed from max_tokens*top_k (3072) but total_padded_slots in run() is num_experts*max_chunks*128 (6144). The buffer was too small, causing index out of bounds.