Fix openmp header (dmlc#3325)

standbyme · Sep 13, 2021 · e7ea0f5 · e7ea0f5
1 parent 26b6318
commit e7ea0f5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/array/cpu/rowwise_pick.h b/src/array/cpu/rowwise_pick.h
@@ -7,7 +7,7 @@
 #define DGL_ARRAY_CPU_ROWWISE_PICK_H_
 
 #include <dgl/array.h>
-#include <omp.h>
+#include <dmlc/omp.h>
 #include <functional>
 #include <algorithm>
 #include <string>