apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2026-04-24 01:29:57 +08:00
Code Issues Actions 11 Packages Projects Releases Wiki Activity

[Fix] Use paddle.device.get_device_properties for multi-platform compatibility (#6400)

Browse Source 
Replace paddle.device.cuda.get_device_properties() with paddle.device.get_device_properties()
to support all hardware platforms (NVIDIA, ILUVATAR, HPU, etc.)

Co-authored-by: Claude Opus 4.5 <noreply@anthropic.com>
This commit is contained in:
Mattheliu
2026-02-09 19:15:41 +08:00
committed by GitHub
parent 35c24f3f71
commit d75b1b8df1
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
fastdeploy/model_executor/ops/gpu/__init__.py
+5 -3
View File
@@ -21,14 +21,16 @@ PACKAGE = "fastdeploy.model_executor.ops.gpu"
def decide_module():
import os
import paddle
prop = paddle.device.cuda.get_device_properties()
# Use paddle.device.get_device_properties() instead of paddle.device.cuda.get_device_properties()
# to support all hardware platforms (NVIDIA, ILUVATAR, HPU, etc.)
prop = paddle.device.get_device_properties()
sm_version = prop.major * 10 + prop.minor
print(f"current sm_version={sm_version}")
import os
curdir = os.path.dirname(os.path.abspath(__file__))
sm_version_path = os.path.join(curdir, f"fastdeploy_ops_{sm_version}")
if os.path.exists(sm_version_path):