File tree Expand file tree Collapse file tree 3 files changed +2
-2
lines changed
patches/FasterTransformer/cuda Expand file tree Collapse file tree 3 files changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -120,7 +120,7 @@ file(TO_NATIVE_PATH ${CMAKE_BINARY_DIR}/${THIRD_PATH}/source/${THIRD_PARTY_NAME}
120
120
file (TO_NATIVE_PATH ${OPS_SOURCE_DIR} /patches/FasterTransformer/CMakeLists.txt cmakelists_src)
121
121
file (TO_NATIVE_PATH ${CMAKE_BINARY_DIR} /${THIRD_PATH} /source /${THIRD_PARTY_NAME} /CMakeLists.txt cmakelists_dst)
122
122
123
- file (TO_NATIVE_PATH ${OPS_SOURCE_DIR} /patches/FasterTransformer/topk_kernels.cu topk_kernels_src)
123
+ file (TO_NATIVE_PATH ${OPS_SOURCE_DIR} /patches/FasterTransformer/cuda/ topk_kernels.cu topk_kernels_src)
124
124
file (TO_NATIVE_PATH ${CMAKE_BINARY_DIR} /${THIRD_PATH} /source /${THIRD_PARTY_NAME} /fastertransformer/cuda/topk_kernels.cu topk_kernels_dst)
125
125
126
126
# TODO(guosheng): `find` seems meeting errors missing argument to `-exec', fix it
Original file line number Diff line number Diff line change @@ -207,7 +207,7 @@ std::vector<std::vector<int64_t>> DecodingInferShape(
207
207
} else if (decoding_strategy == " topk_sampling" ||
208
208
decoding_strategy == " topp_sampling" ) {
209
209
output_dims = {max_len, batch_size};
210
- return {output_dims, {0 }, sequence_length_dims};
210
+ return {output_dims, {1 }, sequence_length_dims};
211
211
} else {
212
212
PD_THROW (" Not supported decoding strategy. " );
213
213
}
File renamed without changes.
You can’t perform that action at this time.
0 commit comments