"docs/vscode:/vscode.git/clone" did not exist on "6cf84866305a5de321e391ef1bba4dbd081dfd41"
Commit 3b9a9a90 authored by rusty1s's avatar rusty1s
Browse files

[skip ci] python 3.9 wheels

parent f9c38ab1
......@@ -10,8 +10,8 @@ jobs:
strategy:
matrix:
os: [macos-10.15, ubuntu-18.04]
python-version: [3.9]
torch-version: [1.8.0]
python-version: [3.9] # [3.6, 3.7, 3.8, 3.9]
torch-version: [1.8.0] # [1.8.0, 1.9.0]
cuda-version: ['cpu', 'cu102', 'cu111']
exclude:
- os: macos-10.15
......
......@@ -34,7 +34,8 @@ def get_extensions():
extra_link_args = ['-s']
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']
if sys.platform == 'win32':
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