Function sample_argmax

ds4.c:22557–22568

            return;
        }
    }
    job->ids_ok = true;
    const ds4_gpu_stream_expert_table table =
        graph_stream_expert_table_make(job->model,
                                       job->layer,
                                       job->il,
                                       job->gate_expert_bytes,
                                       job->down_expert_bytes);
    if (ds4_gpu_stream_expert_cache_begin_selected_load(
                &table,
                job->selected_ids,
                DS4_N_EXPERT_USED) == 0) {
        return;
    }