[GCU] Support gcu platform (#2702)

baseline: e7fa57ebae

Co-authored-by: yongqiangma <xing.wo@163.com>
This commit is contained in:
EnflameGCU
2025-07-08 13:00:52 +08:00
committed by GitHub
parent 26d5d737dd
commit d0f4d6ba3a
33 changed files with 2988 additions and 85 deletions

View File

@@ -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"
]
},