File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
src/libtorchaudio/cuctc/src Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -224,8 +224,6 @@ __global__ void matrix_merge_kernel_v2(
224224 const int batch_id = blockIdx .y ;
225225 if (!log_prob_struct.need_process_on_ith_step (batch_id, step))
226226 return ;
227- const int select_seq =
228- log_prob_struct.ith_selected_seq_in_this_batch (batch_id, step);
229227 __shared__ int tmpclen[128 ]; // beam<128
230228 int tidin, tidout;
231229
@@ -395,7 +393,9 @@ __launch_bounds__(BLOCK_SIZE) void topk_reduce_and_copy_list_per_batch_kernel(
395393 int * clist2,
396394 int blid,
397395 float * score) {
396+ #ifdef USE_PARALLEL_WRITE
398397 constexpr int MAX_SUPPORT_BEAM = 128 ;
398+ #endif
399399 int batch_id = blockIdx .x ;
400400 int rw_offset_this_block = batch_id * items_per_batch;
401401 if (batch_id >= bs)
You can’t perform that action at this time.
0 commit comments