mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2025-12-24 13:28:13 +08:00
[GCU] Support gcu platform (#2702)
baseline: e7fa57ebae
Co-authored-by: yongqiangma <xing.wo@163.com>
This commit is contained in:
4
setup.py
4
setup.py
@@ -167,6 +167,8 @@ def get_device_type():
|
||||
return "npu"
|
||||
elif paddle.is_compiled_with_custom_device('iluvatar_gpu'):
|
||||
return "iluvatar-gpu"
|
||||
elif paddle.is_compiled_with_custom_device('gcu'):
|
||||
return "gcu"
|
||||
else:
|
||||
return "cpu"
|
||||
|
||||
@@ -199,7 +201,7 @@ setup(
|
||||
"model_executor/ops/xpu/libs/*", "model_executor/ops/npu/*",
|
||||
"model_executor/ops/base/*", "model_executor/ops/iluvatar/*",
|
||||
"model_executor/models/*", "model_executor/layers/*",
|
||||
"input/mm_processor/utils/*",
|
||||
"input/mm_processor/utils/*", "model_executor/ops/gcu/*",
|
||||
"version.txt"
|
||||
]
|
||||
},
|
||||
|
||||
Reference in New Issue
Block a user