-
Notifications
You must be signed in to change notification settings - Fork 26.3k
Port ATen/native to ATen/Parallel #20043
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Conversation
Summary: Remove explicit usage of OpenMP from ATen/native Test Plan: BLAS=MKL USE_MKLDNN=1 USE_OPENCV=1 USE_FFMPEG=1 python setup.py develop --cmake pytest -s -v test/test_torch.py::TestTorch
Port ATen/native to ATen/Parallel Summary: Remove explicit usage of OpenMP from ATen/native Test Plan: BLAS=MKL USE_MKLDNN=1 USE_OPENCV=1 USE_FFMPEG=1 python setup.py develop --cmake pytest -s -v test/test_torch.py::TestTorch gh-metadata: pytorch pytorch 20043 gh/ilia-cher/12/head
dzhulgakov
left a comment
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
heroic work!
Port ATen/native to ATen/Parallel Summary: Remove explicit usage of OpenMP from ATen/native Test Plan: BLAS=MKL USE_MKLDNN=1 USE_OPENCV=1 USE_FFMPEG=1 python setup.py develop --cmake pytest -s -v test/test_torch.py::TestTorch gh-metadata: pytorch pytorch 20043 gh/ilia-cher/12/head
|
bugfix + double checked all other call sites |
|
|
||
| #ifdef _OPENMP | ||
| auto parallel_section = [&](int64_t start, int64_t end) { | ||
| for (int64_t i = 0; i < numel; i++) { |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
btw, note that here this is correct, we parallelize inside the loop
dzhulgakov
left a comment
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
ok, I'll trust your audit :)
Port ATen/native to ATen/Parallel Summary: Remove explicit usage of OpenMP from ATen/native Test Plan: BLAS=MKL USE_MKLDNN=1 USE_OPENCV=1 USE_FFMPEG=1 python setup.py develop --cmake pytest -s -v test/test_torch.py::TestTorch gh-metadata: pytorch pytorch 20043 gh/ilia-cher/12/head
Port ATen/native to ATen/Parallel Summary: Remove explicit usage of OpenMP from ATen/native Test Plan: BLAS=MKL USE_MKLDNN=1 USE_OPENCV=1 USE_FFMPEG=1 python setup.py develop --cmake pytest -s -v test/test_torch.py::TestTorch gh-metadata: pytorch pytorch 20043 gh/ilia-cher/12/head
Port ATen/native to ATen/Parallel Summary: Remove explicit usage of OpenMP from ATen/native Test Plan: BLAS=MKL USE_MKLDNN=1 USE_OPENCV=1 USE_FFMPEG=1 python setup.py develop --cmake pytest -s -v test/test_torch.py::TestTorch gh-metadata: pytorch pytorch 20043 gh/ilia-cher/12/head
Port ATen/native to ATen/Parallel Summary: Remove explicit usage of OpenMP from ATen/native Test Plan: BLAS=MKL USE_MKLDNN=1 USE_OPENCV=1 USE_FFMPEG=1 python setup.py develop --cmake pytest -s -v test/test_torch.py::TestTorch gh-metadata: pytorch pytorch 20043 gh/ilia-cher/12/head
Port ATen/native to ATen/Parallel Summary: Remove explicit usage of OpenMP from ATen/native Test Plan: BLAS=MKL USE_MKLDNN=1 USE_OPENCV=1 USE_FFMPEG=1 python setup.py develop --cmake pytest -s -v test/test_torch.py::TestTorch gh-metadata: pytorch pytorch 20043 gh/ilia-cher/12/head
Port ATen/native to ATen/Parallel Summary: Remove explicit usage of OpenMP from ATen/native Test Plan: BLAS=MKL USE_MKLDNN=1 USE_OPENCV=1 USE_FFMPEG=1 python setup.py develop --cmake pytest -s -v test/test_torch.py::TestTorch gh-metadata: pytorch pytorch 20043 gh/ilia-cher/12/head
Port ATen/native to ATen/Parallel Summary: Remove explicit usage of OpenMP from ATen/native Test Plan: BLAS=MKL USE_MKLDNN=1 USE_OPENCV=1 USE_FFMPEG=1 python setup.py develop --cmake pytest -s -v test/test_torch.py::TestTorch gh-metadata: pytorch pytorch 20043 gh/ilia-cher/12/head
Port ATen/native to ATen/Parallel Summary: Remove explicit usage of OpenMP from ATen/native Test Plan: BLAS=MKL USE_MKLDNN=1 USE_OPENCV=1 USE_FFMPEG=1 python setup.py develop --cmake pytest -s -v test/test_torch.py::TestTorch gh-metadata: pytorch pytorch 20043 gh/ilia-cher/12/head
Port ATen/native to ATen/Parallel Summary: Remove explicit usage of OpenMP from ATen/native Test Plan: BLAS=MKL USE_MKLDNN=1 USE_OPENCV=1 USE_FFMPEG=1 python setup.py develop --cmake pytest -s -v test/test_torch.py::TestTorch gh-metadata: pytorch pytorch 20043 gh/ilia-cher/12/head
Port ATen/native to ATen/Parallel Summary: Remove explicit usage of OpenMP from ATen/native Test Plan: BLAS=MKL USE_MKLDNN=1 USE_OPENCV=1 USE_FFMPEG=1 python setup.py develop --cmake pytest -s -v test/test_torch.py::TestTorch gh-metadata: pytorch pytorch 20043 gh/ilia-cher/12/head
Port ATen/native to ATen/Parallel Summary: Remove explicit usage of OpenMP from ATen/native Test Plan: BLAS=MKL USE_MKLDNN=1 USE_OPENCV=1 USE_FFMPEG=1 python setup.py develop --cmake pytest -s -v test/test_torch.py::TestTorch gh-metadata: pytorch pytorch 20043 gh/ilia-cher/12/head
|
@ilia-cher merged this pull request in 2a104f7. |
Summary: Pull Request resolved: pytorch/pytorch#20043 ghimport-source-id: 8003ef8ca335d4c4717a886a3a75fd78ec53ade5 Differential Revision: D15248505 Pulled By: ilia-cher fbshipit-source-id: 7be500ed8bfb23cc36f1dd7108e344319e3e5332
Stack from ghstack:
Summary:
Remove explicit usage of OpenMP from ATen/native
Test Plan:
BLAS=MKL USE_MKLDNN=1 USE_OPENCV=1 USE_FFMPEG=1 python setup.py develop --cmake
pytest -s -v test/test_torch.py::TestTorch
Differential Revision: D15248505