Commit 3b9a9a90 authored by rusty1s's avatar rusty1s
Browse files

[skip ci] python 3.9 wheels

parent f9c38ab1
...@@ -10,8 +10,8 @@ jobs: ...@@ -10,8 +10,8 @@ jobs:
strategy: strategy:
matrix: matrix:
os: [macos-10.15, ubuntu-18.04] os: [macos-10.15, ubuntu-18.04]
python-version: [3.9] python-version: [3.9] # [3.6, 3.7, 3.8, 3.9]
torch-version: [1.8.0] torch-version: [1.8.0] # [1.8.0, 1.9.0]
cuda-version: ['cpu', 'cu102', 'cu111'] cuda-version: ['cpu', 'cu102', 'cu111']
exclude: exclude:
- os: macos-10.15 - os: macos-10.15
......
...@@ -34,7 +34,8 @@ def get_extensions(): ...@@ -34,7 +34,8 @@ def get_extensions():
extra_link_args = ['-s'] extra_link_args = ['-s']
info = parallel_info() info = parallel_info()
if 'backend: OpenMP' in info and 'OpenMP not found' not in info: if ('backend: OpenMP' in info and 'OpenMP not found' not in info
and sys.platform != 'darwin'):
extra_compile_args['cxx'] += ['-DAT_PARALLEL_OPENMP'] extra_compile_args['cxx'] += ['-DAT_PARALLEL_OPENMP']
if sys.platform == 'win32': if sys.platform == 'win32':
extra_compile_args['cxx'] += ['/openmp'] extra_compile_args['cxx'] += ['/openmp']
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment