mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2025-10-04 16:22:57 +08:00
[GCU] Support gcu platform (#2702)
baseline: e7fa57ebae
Co-authored-by: yongqiangma <xing.wo@163.com>
This commit is contained in:
@@ -69,6 +69,12 @@ class Platform:
|
||||
"""
|
||||
return paddle.is_compiled_with_custom_device("iluvatar_gpu")
|
||||
|
||||
def is_gcu(self) -> bool:
|
||||
"""
|
||||
whether platform is gcu
|
||||
"""
|
||||
return paddle.is_compiled_with_custom_device("gcu")
|
||||
|
||||
@classmethod
|
||||
def get_attention_backend_cls(self, selected_backend):
|
||||
"""Get the attention backend"""
|
||||
|
Reference in New Issue
Block a user