Skip to content

Commit 5944cc9

Browse files
authored
Re-organize patches (#621)
* re-organize ops patches
1 parent d336030 commit 5944cc9

File tree

3 files changed

+2
-2
lines changed

3 files changed

+2
-2
lines changed

paddlenlp/ops/CMakeLists.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -120,7 +120,7 @@ file(TO_NATIVE_PATH ${CMAKE_BINARY_DIR}/${THIRD_PATH}/source/${THIRD_PARTY_NAME}
120120
file(TO_NATIVE_PATH ${OPS_SOURCE_DIR}/patches/FasterTransformer/CMakeLists.txt cmakelists_src)
121121
file(TO_NATIVE_PATH ${CMAKE_BINARY_DIR}/${THIRD_PATH}/source/${THIRD_PARTY_NAME}/CMakeLists.txt cmakelists_dst)
122122

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)
124124
file(TO_NATIVE_PATH ${CMAKE_BINARY_DIR}/${THIRD_PATH}/source/${THIRD_PARTY_NAME}/fastertransformer/cuda/topk_kernels.cu topk_kernels_dst)
125125

126126
# TODO(guosheng): `find` seems meeting errors missing argument to `-exec', fix it

paddlenlp/ops/faster_transformer/src/fusion_decoding_op.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -207,7 +207,7 @@ std::vector<std::vector<int64_t>> DecodingInferShape(
207207
} else if (decoding_strategy == "topk_sampling" ||
208208
decoding_strategy == "topp_sampling") {
209209
output_dims = {max_len, batch_size};
210-
return {output_dims, {0}, sequence_length_dims};
210+
return {output_dims, {1}, sequence_length_dims};
211211
} else {
212212
PD_THROW("Not supported decoding strategy. ");
213213
}
File renamed without changes.

0 commit comments

Comments
 (0)