Only specify protobuf version on x86

Change-Id: I71f9f68a37d52bad4ebc34c02f048cdba825eec9
Jeffwhen · Aug 26, 2022 · 6611f1b · 6611f1b
1 parent c7b9edc
commit 6611f1b
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/python/setup.py b/python/setup.py
@@ -32,6 +32,9 @@ def iter_shared_objects():
 packages = ['tpu_perf']
 so_list = list(iter_shared_objects())
 
+if_x86 = any(('x86_64' in arg) for arg in sys.argv)
+protobuf = 'protobuf==3.19.*' if if_x86 else 'protobuf'
+
 setup(
  version=get_version_from_tag(),
  author='sophgo',
@@ -40,7 +43,7 @@ def iter_shared_objects():
  license='Apache',
  name='tpu_perf',
  url='https://www.sophgo.com/',
- install_requires=['numpy', 'lmdb', 'protobuf==3.19.*', 'psutil', 'pyyaml'],
+ install_requires=['numpy', 'lmdb', protobuf, 'psutil', 'pyyaml'],
  packages=find_packages(),
  include_package_data=True,
  package_data={'tpu_perf': so_list})