Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
63 commits
Select commit Hold shift + click to select a range
0378119
fix custom device save error (#75961)
YqGe585 Oct 21, 2025
e106142
fix blas for custom device (#75969)
YqGe585 Oct 21, 2025
011e42d
Revert "Revert "Disable NVIDIA_TF32_OVERRIDE by default for better pr…
A-nnonymous Oct 21, 2025
1f00e21
[Compat] Define the macro `CHECK` only when it is not already defined…
co63oc Oct 21, 2025
3836c2d
[DLPack] Implement dtype and device exchange protocol (#75973)
SigureMo Oct 21, 2025
4b0215a
[CppExtension] Support `os.PathLike` in `CppExtension`/`CUDAExtension…
SigureMo Oct 22, 2025
449eb7e
Support md5 checksum for API output tensor (#75835)
DanielSun11 Oct 22, 2025
d05a775
fix shape=int for size_args_decorator (#75983)
HydrogenSulfate Oct 22, 2025
c46e8c7
fix typo disable_loggling -> disable_logging (#75978)
co63oc Oct 22, 2025
bd5058a
fix _get_arch_info (#75921)
co63oc Oct 22, 2025
7a31a7e
clean some IS_TRT_VERSION_GE(5130) (#75946)
co63oc Oct 22, 2025
05af9c2
clean some IS_TRT_VERSION_GE(8000) (#75944)
co63oc Oct 22, 2025
5f1ea8a
clean some IS_TRT_VERSION_LT(8000) (#75919)
co63oc Oct 22, 2025
c09079f
clean get_cuda_version < 8100 (#75895)
co63oc Oct 22, 2025
19e5388
clean get_cuda_version() < 11020 - part (#75618)
co63oc Oct 22, 2025
4648625
clean get_cuda_version() < 11020 in test_variable_length_memory_effic…
co63oc Oct 22, 2025
ab39b91
clean IS_TRT_VERSION_LT(8000) in tensorrt plugin (#75920)
co63oc Oct 22, 2025
e006936
fix test_dynamic_engine (#75943)
co63oc Oct 22, 2025
c7a658d
[Bug Fix] Fix missing header include in activation_offloader.h (#75936)
youge325 Oct 22, 2025
31870e6
revert_mkl_num_threads (#75985)
zhengshengning Oct 22, 2025
e1ffaed
[Bug Fix] Improve error handling and compatibility in TensorRT engine…
youge325 Oct 22, 2025
05439b1
4th-batch-68-代码梯度计算错误 (#75787)
ApricityXX Oct 22, 2025
3e31bf4
Revert test_activation_op.py to fix bug caused by commit deed9d360d (…
scyyh11 Oct 22, 2025
5ec5c07
4th-batch-19-代码调用错误 (#75759)
ApricityXX Oct 22, 2025
d2f87b7
4th-batch-17-代码限制多设备场景(补充修复) (#75959)
ApricityXX Oct 22, 2025
2bb1097
【UnitTestFix No.3】fix test_conv3d_transpose_op.py (#75945)
scyyh11 Oct 22, 2025
a84cc0e
[Bug Fix] add missing header include in ir_context.h (#75927)
youge325 Oct 22, 2025
1f84292
add tensorrt 10 support int64 (#75951)
co63oc Oct 22, 2025
2b9ba85
[Compat] Try import `tvm_ffi` when enable torch proxy (#75991)
SigureMo Oct 22, 2025
89931f0
clean pip3.8 in Dockerfile.develop.npu (#75893)
co63oc Oct 23, 2025
ff34cae
fix masked_fill_grad value_grad bug (#75988)
changeyoung98 Oct 23, 2025
4263da4
4th-batch-20-代码存在未被使用的变量 (#75761)
ApricityXX Oct 23, 2025
b65dadd
use op_test.get_cuda_version (#75994)
co63oc Oct 23, 2025
89d92c3
merge ifdef PADDLE_WITH_CUDA in build_strategy.cc (#75962)
co63oc Oct 23, 2025
6692ccb
[Cherry-pick] Optimize FlashMask v3 performance (#75737) (#75984)
umiswing Oct 23, 2025
481a88f
[Stride] Disable Split Stride Kernel (#75987)
Eddie-Wang1120 Oct 23, 2025
9fe6225
[Bug Fix] Fix NaN/Inf check to support float16, bfloat16, and complex…
youge325 Oct 23, 2025
74f6ea8
[Stride] Optimizing H2D Copy by TensorIterator and OpenMP (#75192)
Eddie-Wang1120 Oct 23, 2025
83d4454
[Precision Depth Alignment] implement torch compatible max_pool2d gra…
zrr1999 Oct 23, 2025
189706c
fix to_tensor bug (#76000)
wanghuancoder Oct 23, 2025
2db3061
[CINN] Fix bug of infer_symbol_shape for crop op (#75992)
zyfncg Oct 23, 2025
ca3f6ef
【CUDA Kernel No.93】psroi_pool_grad_kernel算子修复 (#75938)
xxiu1 Oct 23, 2025
246c4a9
fix win32 rms_norm. (#76007)
A-nnonymous Oct 23, 2025
c85fc97
Update check_approval.sh (#76012)
luotao1 Oct 24, 2025
a799f8d
[Fix] log sigmoid complex (#75953)
scyyh11 Oct 24, 2025
7d3ae36
[PHI] Flash Attention V3 128B aligned chunking load/store (#76003)
Enigmatisms Oct 24, 2025
fafb525
[Slice] Fix big tensor (#76004)
Eddie-Wang1120 Oct 24, 2025
cdfc18c
fix python version in ci/utils.sh (#75997)
co63oc Oct 24, 2025
ed8e5e3
clean pip3.8 in Dockerfile.develop.dtk (#75738)
co63oc Oct 24, 2025
eee3605
fix repeat IS_TRT_VERSION_GE (#75975)
co63oc Oct 24, 2025
f0747d3
clean IS_TRT_VERSION_GE(5000) (#75990)
co63oc Oct 24, 2025
c4a227e
Initial plan
Copilot Oct 24, 2025
9b7c930
Fix int32 overflow in elementwise_grad_kernel_impl.h
Copilot Oct 24, 2025
759de4d
Fix int32 overflow in accuracy_check and isclose kernel impl
Copilot Oct 24, 2025
7d64474
Fix int32 overflow in renorm, unstack, kldiv, and svdvals_grad impl
Copilot Oct 24, 2025
22d5f37
Fix int32 overflow in gumbel_softmax and kldiv_loss impl
Copilot Oct 24, 2025
ea2ba17
Fix int32 overflow in lrn and frame kernel impl
Copilot Oct 24, 2025
c9770ce
Fix function signatures in lrn_kernel_impl to match int64_t parameters
Copilot Oct 24, 2025
f1d21ba
Add validation checks for large tensor support in LRN kernels
Copilot Oct 26, 2025
91b6c2f
Fix int32 overflow in stft and fold/unfold kernel impl
Copilot Oct 27, 2025
ef6cb1a
Fix int32 overflow in lstm, lstsq, qr_grad, and spectral_norm_grad impl
Copilot Oct 27, 2025
80e2535
Fix int32 overflow in warpctc, warprnnt, gru_unit and spectral_norm impl
Copilot Oct 27, 2025
1b431b8
Fix int32 overflow in svd_grad and conv kernel impl
Copilot Oct 27, 2025
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 6 additions & 0 deletions .github/workflows/docker.yml
Original file line number Diff line number Diff line change
Expand Up @@ -103,6 +103,12 @@ jobs:
for name in "${!docker_files[@]}"
do
md5_value=`md5sum tools/dockerfile/${docker_files[$name]} | awk '{print $1}'`
if [ $name == "docker_dcu" ]; then
md5_value="76937a563116f6008c8ca4cb4f592759"
fi
if [ $name == "docker_npu" ]; then
md5_value="a3793bdeea5ae881a0c1eaf4d7c30c64"
fi
docker_image="ccr-2vdh3abv-pub.cnc.bj.baidubce.com/ci/paddle:${md5_value}"
declare "${name}_image=${docker_image}"
echo "${name}_image=${docker_image}" >> $GITHUB_OUTPUT
Expand Down
48 changes: 24 additions & 24 deletions ci/check_approval.sh
Original file line number Diff line number Diff line change
Expand Up @@ -55,8 +55,8 @@ function run_tools_test() {

changed_env_var_count=`git diff -U0 upstream/$BRANCH ${PADDLE_ROOT}/paddle | grep 'DEFINE_EXPORTED' | grep -v '@@' | wc -l`
if [[ $changed_env_var_count -gt 0 ]]; then
echo_line="You must have one RD (phlrain or luotao1) approval for changing the FLAGS, which manages the environment variables.\n"
check_approval 1 phlrain luotao1
echo_line="You must have one RD (phlrain) approval for changing the FLAGS, which manages the environment variables.\n"
check_approval 1 phlrain
fi

changed_deprecated_tests_count=$(expr $(git ls-tree -r --name-only HEAD ${PADDLE_ROOT}/test/deprecated | grep '^test' | wc -l) - $(git ls-tree -r --name-only upstream/$BRANCH ${PADDLE_ROOT}/test/deprecated | grep '^tes' | wc -l))
Expand Down Expand Up @@ -116,8 +116,8 @@ fi

HAS_PADDLE_GET=`git diff -U0 upstream/$BRANCH $FILTER |grep "^+" |grep -o -m 1 "paddle::get" || true`
if [ ${HAS_PADDLE_GET} ] && [ "${PR_ID}" != "" ]; then
echo_line="paddle::get is not recommended for direct use, because it may throw an bad_variant_access exception without any stack information, so please use PADDLE_GET(_**)(dtype, value) series macros here. If these macros cannot meet your needs, please use try-catch to handle paddle::get and request luotao1 or zhangbo9674 or phlrain review and approve.\n"
check_approval 1 luotao1 zhangbo9674 phlrain
echo_line="paddle::get is not recommended for direct use, because it may throw an bad_variant_access exception without any stack information, so please use PADDLE_GET(_**)(dtype, value) series macros here. If these macros cannot meet your needs, please use try-catch to handle paddle::get and request zhangbo9674 or phlrain review and approve.\n"
check_approval 1 zhangbo9674 phlrain
fi

HAS_LEGACY_KERNEL_REGISTRATION=`git diff -U0 upstream/$BRANCH $FILTER | grep '^\+' | grep -oE -m 1 "REGISTER_OP[A-Z_]{1,9}KERNEL[_FUNCTOR|_WITH_CUSTOM_TYPE|_EX]*" || true`
Expand Down Expand Up @@ -170,8 +170,8 @@ fi
NO_NPU_FILE=`git diff --name-only upstream/$BRANCH | grep -v "_npu.py"`
HAS_UNITTEST_SKIP=`git diff -U0 upstream/$BRANCH ${NO_NPU_FILE} | grep "^+[[:space:]]\{0,\}@unittest.skip" || true`
if [ "${HAS_UNITTEST_SKIP}" != "" ] && [ "${PR_ID}" != "" ]; then
echo_line="Unittest is not allowed to be disabled.\nYou must have one RD (kolinwei(Recommend), wanghuancoder, luotao1, QingshuChen) approval for the usage of @unittest.skip or @unittest.skipIf.\n${HAS_UNITTEST_SKIP}\n"
check_approval 1 kolinwei wanghuancoder luotao1 QingshuChen
echo_line="Unittest is not allowed to be disabled.\nYou must have one RD (kolinwei(Recommend), wanghuancoder, QingshuChen) approval for the usage of @unittest.skip or @unittest.skipIf.\n${HAS_UNITTEST_SKIP}\n"
check_approval 1 kolinwei wanghuancoder QingshuChen
fi

HAS_MODIFIED_DEMO_CMAKE=`git diff --name-only upstream/$BRANCH | grep "paddle/fluid/inference/api/demo_ci/CMakeLists.txt" || true`
Expand Down Expand Up @@ -332,8 +332,8 @@ fi

INVALID_UNITTEST_ASSERT_CHECK=`echo "$ALL_ADDED_LINES" | grep -zoE '\+\s+((assert\s+)|(self\.assert(True|Equal)\())(\s*\+\s*)?(np|numpy)\.(allclose|array_equal)[^+]*' || true`
if [ "${INVALID_UNITTEST_ASSERT_CHECK}" != "" ] && [ "${PR_ID}" != "" ]; then
echo_line="It is recommended to use 'np.testing.assert_allclose' and 'np.testing.assert_array_equal' instead of 'self.assertTrue(np.allclose(...))' and 'self.assertTrue(np.array_equal(...))'.\nPlease modify the code below. If anything is unclear, please read the specification [ https://github.com/PaddlePaddle/community/blob/master/rfcs/CodeStyle/20220805_code_style_improvement_for_unittest.md#background ]. If it is a mismatch, please request SigureMo (Recommend) or zrr1999 or luotao1 review and approve.\nThe code that do not meet the specification are as follows:\n${INVALID_UNITTEST_ASSERT_CHECK}\n"
check_approval 1 SigureMo zrr1999 luotao1
echo_line="It is recommended to use 'np.testing.assert_allclose' and 'np.testing.assert_array_equal' instead of 'self.assertTrue(np.allclose(...))' and 'self.assertTrue(np.array_equal(...))'.\nPlease modify the code below. If anything is unclear, please read the specification [ https://github.com/PaddlePaddle/community/blob/master/rfcs/CodeStyle/20220805_code_style_improvement_for_unittest.md#background ]. If it is a mismatch, please request SigureMo (Recommend) or zrr1999 review and approve.\nThe code that do not meet the specification are as follows:\n${INVALID_UNITTEST_ASSERT_CHECK}\n"
check_approval 1 SigureMo zrr1999
fi

TEST_FILE_ADDED_LINES=$(git diff -U0 upstream/$BRANCH -- test |grep "^+")
Expand Down Expand Up @@ -459,21 +459,21 @@ if [ "${NEW_OP_ADDED}" != "" ] && [ "${PR_ID}" != "" ]; then
GET_KERNEL_TYPE_FUNC_CNT=`git diff -U0 --diff-filter=A upstream/$BRANCH |grep "+" |grep -czoE "GetExpectedKernelType[(][^(){}]+[)][^{]+[{][^}]+[}]" || true`
INDICATE_VAR_DTYPE_CNT=`git diff -U0 --diff-filter=A upstream/$BRANCH |grep "+" |grep -co "IndicateVarDataType" || true`
if [ ${GET_KERNEL_TYPE_FUNC_CNT} -gt ${INDICATE_VAR_DTYPE_CNT} ]; then
echo_line="If you override GetExpectedKernelType method of OperatorWithKernel, please use OperatorWithKernel::IndicateVarDataType() method to get specific input variable's dtype, which checked whether the input variable is initialized (The details in https://github.com/PaddlePaddle/FluidDoc/pull/1527). If you don't use this method to check, you must have one RD (zhangbo9674 or phlrain or luotao1) approval for the usage of other methods.\n"
check_approval 1 luotao1 zhangbo9674 phlrain
echo_line="If you override GetExpectedKernelType method of OperatorWithKernel, please use OperatorWithKernel::IndicateVarDataType() method to get specific input variable's dtype, which checked whether the input variable is initialized (The details in https://github.com/PaddlePaddle/FluidDoc/pull/1527). If you don't use this method to check, you must have one RD (zhangbo9674 or phlrain ) approval for the usage of other methods.\n"
check_approval 1 zhangbo9674 phlrain
fi
fi

HAS_OPERATORBASE_FLAG=`git diff -U0 --diff-filter=A upstream/$BRANCH | grep -E "public[[:space:]]+.*OperatorBase" || true`
if [ "${HAS_OPERATORBASE_FLAG}" != "" ] && [ "${PR_ID}" != "" ]; then
echo_line="In order to support dynamic graph, all ops are not recommended to inherit OperatorBase. Please use OperatorWithKernel instead.\nYou must have one RD (phlrain (Recommend), luotao1, XiaoguangHu01) approval for the inherit of OperatorBase.\nYou inherit the OperatorBase class. The corresponding lines are as follows:\n${HAS_OPERATORBASE_FLAG}"
check_approval 1 phlrain luotao1 XiaoguangHu01
echo_line="In order to support dynamic graph, all ops are not recommended to inherit OperatorBase. Please use OperatorWithKernel instead.\nYou must have one RD (phlrain (Recommend), XiaoguangHu01) approval for the inherit of OperatorBase.\nYou inherit the OperatorBase class. The corresponding lines are as follows:\n${HAS_OPERATORBASE_FLAG}"
check_approval 1 phlrain XiaoguangHu01
fi

HAS_INPLACE_TESTS=`git diff -U0 upstream/$BRANCH |grep "+" |grep -E "inplace_atol[[:space:]]*=.*" || true`
if [ "${HAS_INPLACE_TESTS}" != "" ] && [ "${PR_ID}" != "" ]; then
echo_line="The calculation results of setting inplace enabled and disabled must be equal, that is, it's not recommended to set inplace_atol.\n If you do need to use inplace_atol, you must have one RD (XiaoguangHu01, phlrain, luotao1, QingshuChen) approval for the usage of inplace_atol.\nThe corresponding lines are as follows:\n${HAS_INPLACE_TESTS}\n"
check_approval 1 XiaoguangHu01 phlrain luotao1 QingshuChen
echo_line="The calculation results of setting inplace enabled and disabled must be equal, that is, it's not recommended to set inplace_atol.\n If you do need to use inplace_atol, you must have one RD (XiaoguangHu01, phlrain, QingshuChen) approval for the usage of inplace_atol.\nThe corresponding lines are as follows:\n${HAS_INPLACE_TESTS}\n"
check_approval 1 XiaoguangHu01 phlrain QingshuChen
fi

OP_FILE_CHANGED=`git diff --name-only --diff-filter=AMR upstream/$BRANCH |grep -oE ".+_op..*" || true`
Expand All @@ -488,8 +488,8 @@ if [ "${OP_FILE_CHANGED}" != "" ] && [ "${PR_ID}" != "" ]; then
done
if [ "${ERROR_LINES}" != "" ]; then
ERROR_LINES=${ERROR_LINES//+/'\n+\t'}
echo_line="Using ShareDataWith or ShareBufferWith is not recommended. You must have one RD's (zhhsplendid (Recommend), zhiqiu or luotao1) approval to use these methods. For more information, please refer to https://github.com/PaddlePaddle/Paddle/wiki/ShareDataWith-is-prohibited-in-OP. The error lines are as follows:${ERROR_LINES}"
check_approval 1 zhhsplendid zhiqiu luotao1
echo_line="Using ShareDataWith or ShareBufferWith is not recommended. You must have one RD's (zhhsplendid (Recommend), zhiqiu) approval to use these methods. For more information, please refer to https://github.com/PaddlePaddle/Paddle/wiki/ShareDataWith-is-prohibited-in-OP. The error lines are as follows:${ERROR_LINES}"
check_approval 1 zhhsplendid zhiqiu
fi
fi

Expand All @@ -505,8 +505,8 @@ if [ "${CMAKE_FILE_CHANGED}" != "" ] && [ "${PR_ID}" != "" ]; then
done
if [ "${ERROR_LINES}" != "" ]; then
ERROR_LINES=${ERROR_LINES//+/'\n+\t'}
echo_line="Change compilation flag of warnings is not recommended. You must have one RD's (zhiqiu (Recommend), luotao1 or phlrain) approval to use these methods. "
check_approval 1 zhiqiu luotao1 phlrain
echo_line="Change compilation flag of warnings is not recommended. You must have one RD's (zhiqiu (Recommend), phlrain) approval to use these methods. "
check_approval 1 zhiqiu phlrain
fi
fi

Expand All @@ -519,8 +519,8 @@ if [ "${NEW_OP_TEST_ADDED}" != "" ] && [ "${PR_ID}" != "" ]; then
CHECK_WHOLE=$CHECK_OUTPUT$CHECK_OUTPUT_WITH_PLACE$CHECK_GRAD$CHECK_GRAD_CHECK
if [ "${CHECK_WHOLE}" != "" ] ; then
CHECK_OP=${CHECK_WHOLE//+/'\n+'}
echo_line="Please use the default precision parameters of 'atol, rtol, eps, max_relative_error'. If you don't use the default value, you must have one RD (Xreki (Recommend), QingshuChen(Recommend for kunlun), zhiqiu, luotao1, phlrain or ZzSean) approval for the usage of other values. The detailed information is in the link: https://github.cor/PaddlePaddle/Paddle/wiki/OP-test-accuracy-requirements. The error line is ${CHECK_OP}\n"
check_approval 1 Xreki QingshuChen zhiqiu luotao1 phlrain
echo_line="Please use the default precision parameters of 'atol, rtol, eps, max_relative_error'. If you don't use the default value, you must have one RD (Xreki (Recommend), QingshuChen(Recommend for kunlun), zhiqiu, phlrain or ZzSean) approval for the usage of other values. The detailed information is in the link: https://github.cor/PaddlePaddle/Paddle/wiki/OP-test-accuracy-requirements. The error line is ${CHECK_OP}\n"
check_approval 1 Xreki QingshuChen zhiqiu phlrain
fi
fi

Expand All @@ -536,8 +536,8 @@ if [ "${UNITTEST_FILE_CHANGED}" != "" ] && [ "${PR_ID}" != "" ]; then
done
if [ "${ERROR_LINES}" != "" ]; then
ERROR_LINES=${ERROR_LINES//+/'\n+\t'}
echo_line="It is an Op accuracy problem, please take care of it. You must have one RD (zhangting2020 (Recommend), luotao1 or phlrain, QingshuChen) approval for the usage (either add or delete) of @skip_check_grad_ci. For more information, please refer to: https://github.com/PaddlePaddle/Paddle/wiki/Gradient-Check-Is-Required-for-Op-Test. The corresponding lines are as follows:\n${ERROR_LINES}\n"
check_approval 1 zhangting2020 luotao1 phlrain QingshuChen
echo_line="It is an Op accuracy problem, please take care of it. You must have one RD (zhangting2020 (Recommend), or phlrain, QingshuChen) approval for the usage (either add or delete) of @skip_check_grad_ci. For more information, please refer to: https://github.com/PaddlePaddle/Paddle/wiki/Gradient-Check-Is-Required-for-Op-Test. The corresponding lines are as follows:\n${ERROR_LINES}\n"
check_approval 1 zhangting2020 phlrain QingshuChen
fi
fi

Expand Down Expand Up @@ -586,12 +586,12 @@ UNITYBUILD_RULE_CHANGED=$(git diff --name-only upstream/$BRANCH |
grep "unity_build_rule.cmake" || true)
if [ -n "${UNITYBUILD_RULE_CHANGED}" -a -n "${PR_ID}" ]; then
echo_line="You must have one RD (Avin0323(Recommend) or zhwesky2010 or
wanghuancoder or luotao1) approval for modifying
wanghuancoder) approval for modifying
unity_build_rule.cmake which the rules of Unity Build."
echo_line=$(echo ${echo_line})
# Avin0323(23427135) zhwesky2010(52485244)
# wanghuancoder(26922892) luotao1(6836917)
check_approval 1 Avin0323 zhwesky2010 wanghuancoder luotao1
check_approval 1 Avin0323 zhwesky2010 wanghuancoder
fi

if [ -n "${echo_list}" ];then
Expand Down
37 changes: 5 additions & 32 deletions ci/utils.sh
Original file line number Diff line number Diff line change
Expand Up @@ -297,19 +297,7 @@ function cmake_base() {
SYSTEM=`uname -s`
if [ "$SYSTEM" == "Darwin" ]; then
echo "Using python abi: $1"
if [ "$1" == "cp38-cp38" ]; then
if [ -d "/Library/Frameworks/Python.framework/Versions/3.8" ]; then
export LD_LIBRARY_PATH=/Library/Frameworks/Python.framework/Versions/3.8/lib/
export DYLD_LIBRARY_PATH=${DYLD_LIBRARY_PATH}:/Library/Frameworks/Python.framework/Versions/3.8/lib/
export PATH=/Library/Frameworks/Python.framework/Versions/3.8/bin/:${PATH}
PYTHON_FLAGS="-DPYTHON_EXECUTABLE:FILEPATH=/Library/Frameworks/Python.framework/Versions/3.8/bin/python3
-DPYTHON_INCLUDE_DIR:PATH=/Library/Frameworks/Python.framework/Versions/3.8/include/python3.8/
-DPYTHON_LIBRARY:FILEPATH=/Library/Frameworks/Python.framework/Versions/3.8/lib/libpython3.8.dylib"
pip3.8 install --user -r ${PADDLE_ROOT}/python/requirements.txt
else
exit 1
fi
elif [ "$1" == "cp39-cp39" ]; then
if [ "$1" == "cp39-cp39" ]; then
if [ -d "/Library/Frameworks/Python.framework/Versions/3.9" ]; then
export LD_LIBRARY_PATH=/Library/Frameworks/Python.framework/Versions/3.9/lib/
export DYLD_LIBRARY_PATH=${DYLD_LIBRARY_PATH}:/Library/Frameworks/Python.framework/Versions/3.9/lib/
Expand Down Expand Up @@ -373,15 +361,7 @@ function cmake_base() {
else
if [ "$1" != "" ]; then
echo "using python abi: $1"
if [ "$1" == "cp38-cp38" ]; then
export LD_LIBRARY_PATH=/opt/_internal/cpython-3.8.0/lib/:${LD_LIBRARY_PATH}
export PATH=/opt/_internal/cpython-3.8.0/bin/:${PATH}
export PYTHON_FLAGS="-DPYTHON_EXECUTABLE:FILEPATH=/opt/_internal/cpython-3.8.0/bin/python3.8
-DPYTHON_INCLUDE_DIR:PATH=/opt/_internal/cpython-3.8.0/include/python3.8
-DPYTHON_LIBRARIES:FILEPATH=/opt/_internal/cpython-3.8.0/lib/libpython3.so"
pip3.8 install -r ${PADDLE_ROOT}/python/requirements.txt
pip3.8 install -r ${PADDLE_ROOT}/paddle/scripts/compile_requirements.txt
elif [ "$1" == "cp39-cp39" ]; then
if [ "$1" == "cp39-cp39" ]; then
export LD_LIBRARY_PATH=/opt/_internal/cpython-3.9.0/lib/:${LD_LIBRARY_PATH}
export PATH=/opt/_internal/cpython-3.9.0/bin/:${PATH}
export PYTHON_FLAGS="-DPYTHON_EXECUTABLE:FILEPATH=/opt/_internal/cpython-3.9.0/bin/python3.9
Expand Down Expand Up @@ -413,21 +393,14 @@ function cmake_base() {
-DPYTHON_LIBRARIES:FILEPATH=/opt/_internal/cpython-3.12.0/lib/libpython3.so"
pip3.12 install -r ${PADDLE_ROOT}/python/requirements.txt
pip3.12 install -r ${PADDLE_ROOT}/paddle/scripts/compile_requirements.txt
elif [ "$1" == "cp313-cp313" ]; then
elif [ "$1" == "cp313-cp313" ]; then
export LD_LIBRARY_PATH=/opt/_internal/cpython-3.13.0/lib/:${LD_LIBRARY_PATH}
export PATH=/opt/_internal/cpython-3.13.0/bin/:${PATH}
export PYTHON_FLAGS="-DPYTHON_EXECUTABLE:FILEPATH=/opt/_internal/cpython-3.13.0/bin/python3.13
-DPYTHON_INCLUDE_DIR:PATH=/opt/_internal/cpython-3.13.0/include/python3.13
-DPYTHON_LIBRARIES:FILEPATH=/opt/_internal/cpython-3.13.0/lib/libpython3.so"
pip3.13 install -r ${PADDLE_ROOT}/python/requirements.txt
pip3.13 install -r ${PADDLE_ROOT}/paddle/scripts/compile_requirements.txt
elif [ "$1" == "conda-python3.8" ]; then
export LD_LIBRARY_PATH=/opt/conda/lib/:${LD_LIBRARY_PATH}
export PATH=/opt/conda/bin/:${PATH}
export PYTHON_FLAGS="-DPYTHON_EXECUTABLE:FILEPATH=/opt/conda/bin/python
-DPYTHON_INCLUDE_DIR:PATH=/opt/conda/include/python3.8m
-DPYTHON_LIBRARIES:FILEPATH=/opt/conda/lib/libpython3.so"
/opt/conda/bin/pip install -r ${PADDLE_ROOT}/python/requirements.txt
fi
# for CINN, to find libcuda.so.1
export LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:/usr/local/cuda-11.2/compat/
Expand Down Expand Up @@ -484,7 +457,7 @@ function cmake_base() {
-DCMAKE_EXPORT_COMPILE_COMMANDS=ON
-DWITH_INFERENCE_API_TEST=${WITH_INFERENCE_API_TEST:-ON}
-DINFERENCE_DEMO_INSTALL_DIR=${INFERENCE_DEMO_INSTALL_DIR}
-DPY_VERSION=${PY_VERSION:-3.8}
-DPY_VERSION=${PY_VERSION:-3.9}
-DCMAKE_INSTALL_PREFIX=${INSTALL_PREFIX:-/paddle/build}
-DWITH_PSCORE=${pscore_flag}
-DWITH_PSLIB=${pslib_flag}
Expand Down Expand Up @@ -537,7 +510,7 @@ EOF
-DCMAKE_EXPORT_COMPILE_COMMANDS=ON \
-DWITH_INFERENCE_API_TEST=${WITH_INFERENCE_API_TEST:-ON} \
-DINFERENCE_DEMO_INSTALL_DIR=${INFERENCE_DEMO_INSTALL_DIR} \
-DPY_VERSION=${PY_VERSION:-3.8} \
-DPY_VERSION=${PY_VERSION:-3.9} \
-DCMAKE_INSTALL_PREFIX=${INSTALL_PREFIX:-/paddle/build} \
-DWITH_PSCORE=${pscore_flag} \
-DWITH_PSLIB=${pslib_flag} \
Expand Down
1 change: 1 addition & 0 deletions cmake/configure.cmake
Original file line number Diff line number Diff line change
Expand Up @@ -179,6 +179,7 @@ if(WITH_MKLML AND MKLML_IOMP_LIB)
set(OPENMP_FLAGS "")
else()
set(OPENMP_FLAGS "-fopenmp")
add_definitions(-DPADDLE_WITH_OPENMP)
endif()
set(CMAKE_C_CREATE_SHARED_LIBRARY_FORBIDDEN_FLAGS ${OPENMP_FLAGS})
set(CMAKE_CXX_CREATE_SHARED_LIBRARY_FORBIDDEN_FLAGS ${OPENMP_FLAGS})
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -167,13 +167,11 @@ class FusionOpPattern : public pir::OpRewritePattern<cinn::dialect::FusionOp> {

const std::unordered_map<std::string, CinnOpHandler>& op_handler_map() const {
static std::unordered_map<std::string, CinnOpHandler> handler_map = {
{cinn::dialect::ReshapeOp::name(), &FusionOpPattern::ReshapeOpPattern},
{paddle::dialect::AssignOut_Op::name(),
&FusionOpPattern::AssignOutOpPattern},
{paddle::dialect::CastOp::name(), &FusionOpPattern::CastOpPattern},
#if defined(PADDLE_WITH_HIP)
{cinn::dialect::ConcatOp::name(), &FusionOpPattern::ConcatOpPattern},
#endif
{cinn::dialect::ReshapeOp::name(), &FusionOpPattern::ReshapeOpPattern},
{paddle::dialect::AssignOut_Op::name(),
&FusionOpPattern::AssignOutOpPattern},
{paddle::dialect::CastOp::name(), &FusionOpPattern::CastOpPattern},
{cinn::dialect::ConcatOp::name(), &FusionOpPattern::ConcatOpPattern},
};
return handler_map;
}
Expand Down
Loading