mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2025-12-24 13:28:13 +08:00
dcu adapter ernie45t (#2756)
Co-authored-by: lifu <lifu@sugon.com> Co-authored-by: yongqiangma <xing.wo@163.com>
This commit is contained in:
2
setup.py
2
setup.py
@@ -146,6 +146,8 @@ def load_requirements():
|
||||
requirements_file_name = 'requirements.txt'
|
||||
if paddle.is_compiled_with_custom_device('iluvatar_gpu'):
|
||||
requirements_file_name = 'requirements_iluvatar.txt'
|
||||
elif paddle.is_compiled_with_rocm():
|
||||
requirements_file_name = 'requirements_dcu.txt'
|
||||
requirements_path = os.path.join(os.path.dirname(__file__),
|
||||
requirements_file_name)
|
||||
with open(requirements_path, 'r') as f:
|
||||
|
||||
Reference in New Issue
Block a user