src/nv-wavenet/nv_wavenet_dualblock.cuh (3 lines): - line 185: if (threadIdx.x < R) { //TODO: change to S - line 187: for (int tile_m = 0; tile_m < S/S; tile_m++) { //TODO: change to S - line 194: loadWeights(weights, params.WskipInit + tile_m*S, tile_k, threadIdx.x, S); //TODO: change to S