apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-12-24 13:28:13 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

[Others]Prevent core dumps during Paddle version check (#5657)

Browse Source
This commit is contained in:
bukejiyu
2025-12-23 13:57:45 +08:00
committed by GitHub
parent 9da89a374b
commit 6c36a17369
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
fastdeploy/model_executor/utils.py
View File

@@ -351,6 +351,9 @@ def is_paddle_support_new_h2d():
code = """
import paddle
import resource
resource.setrlimit(resource.RLIMIT_CORE, (0, 0))
try:
dst = paddle.zeros([2, 4], dtype='bfloat16')
src = paddle.ones([2, 2], dtype='bfloat16', device='cpu')