Fix openmp header (#3325)

BarclayII · BarclayII · commit 582f71a173b6 · 2021-11-05T05:13:11.000Z
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>
