Skip to content

Commit

Permalink
Merge pull request #606 from HEXRD/tbb-intel-only
Browse files Browse the repository at this point in the history
Remove tbb requirement for non-intel machines
  • Loading branch information
psavery authored Jan 23, 2024
2 parents 29e7c6f + f37cde9 commit 30f10c0
Showing 1 changed file with 6 additions and 1 deletion.
7 changes: 6 additions & 1 deletion setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@
import distutils.ccompiler
import os
from pathlib import Path
import platform
from setuptools import setup, find_packages, Extension
import subprocess
import sys
Expand All @@ -24,11 +25,15 @@
'scikit-image',
'scikit-learn',
'scipy',
'tbb',
'tqdm',
'xxhash',
]

if platform.machine() == 'x86_64':
# tbb is an optional dependency, but it is nice to have
# Add it to the requirements automatically for intel computers.
install_reqs.append('tbb')

# This will determine which compiler is being used to build the C modules
compiler_type = distutils.ccompiler.get_default_compiler()
if compiler_type in ("unix", "mingw32"):
Expand Down

0 comments on commit 30f10c0

Please sign in to comment.