host_top_k_extremes cost (per call, 200 iters after 20 warmup)
B=  64  0.0680 ms/call
B= 122  0.0723 ms/call
B= 512  0.0982 ms/call
