Work around OMP barrier issue with MSVC (pytorch#2918)

Summary: Pull Request resolved: pytorch#2918 Differential Revision: D60601583 Pulled By: q10
q10 · Aug 1, 2024 · df77f14 · df77f14
1 parent e3db758
commit df77f14
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/fbgemm_gpu/requirements.txt b/fbgemm_gpu/requirements.txt
@@ -18,6 +18,7 @@ jinja2
 mpmath==1.3.0
 ninja
 numpy
+pyre-extensions
 scikit-build
 setuptools
 setuptools_git_versioning

diff --git a/src/Utils.cc b/src/Utils.cc
@@ -801,6 +801,7 @@ std::pair<K*, V*> radix_sort_parallel(
       std::swap(input_keys, output_keys);
       std::swap(input_values, output_values);
 #pragma omp barrier
+      {}
     }
   }
 #ifdef _MSC_VER