适配rocm

10a14e6b · xiabo · f1f53d0f · 10a14e6b
Commit 10a14e6b authored Sep 05, 2023 by xiabo
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

setup.py setup.py +4 -0

No files found.
--- a/setup.py
+++ b/setup.py
@@ -4,6 +4,10 @@ import platform
 import re
 from pkg_resources import DistributionNotFound, get_distribution, parse_version
 from setuptools import find_packages, setup
+import subprocess
+from typing import Optional, Union
+from pathlib import Path
+

 EXT_TYPE = ''
 try: