Skip to content

Commit 48fd1f1

Browse files
q10facebook-github-bot
authored andcommitted
Work around OMP barrier issue with MSVC (pytorch#2918)
Summary: X-link: facebookresearch/FBGEMM#28 Pull Request resolved: pytorch#2918 Reviewed By: sryap Differential Revision: D60601583 Pulled By: q10 fbshipit-source-id: b8b5d244748db26a2591f65c61adc30ebc800540
1 parent 3a5bb1d commit 48fd1f1

File tree

2 files changed

+2
-0
lines changed

2 files changed

+2
-0
lines changed

fbgemm_gpu/requirements.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@ jinja2
1818
mpmath==1.3.0
1919
ninja
2020
numpy
21+
pyre-extensions
2122
scikit-build
2223
setuptools
2324
setuptools_git_versioning

src/Utils.cc

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -801,6 +801,7 @@ std::pair<K*, V*> radix_sort_parallel(
801801
std::swap(input_keys, output_keys);
802802
std::swap(input_values, output_values);
803803
#pragma omp barrier
804+
{}
804805
}
805806
}
806807
#ifdef _MSC_VER

0 commit comments

Comments
 (0)