dcu adapter ernie45t (#2756)

Co-authored-by: lifu <lifu@sugon.com>
Co-authored-by: yongqiangma <xing.wo@163.com>
This commit is contained in:
lifulll
2025-07-09 18:56:27 +08:00
committed by GitHub
parent 03a74995b8
commit 1f28bdf994
30 changed files with 1133 additions and 41 deletions

View File

@@ -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: