Chunked Prefill Chunk Count

Problem

Count the chunks a prompt of seq_len tokens is split into for chunked prefill with chunk_size tokens per chunk.

The signature is chunked_prefill_chunks(seq_len, chunk_size). Return an integer; a trailing partial chunk still counts.

Hint

Ceiling division counts the final partial chunk.

case 1

in: [1000,256]

expected: 4

case 2

in: [512,256]

expected: 2

case 3

in: [100,128]

expected: 1

case 4

in: [1,1]

expected: 1

Starter codePython

def chunked_prefill_chunks(seq_len, chunk_size):
    # Your code here
    pass

4 test cases · run your code against them in the editor