Merge pull request #2 from luwei0917/patch-1

Update find_ranges_grid_ds.cuh
phermosilla · Jan 24, 2022 · cc067a2 · cc067a2
2 parents 9c79ea0 + 96d3b73
commit cc067a2
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/IEProtLib/tf_ops/cu/header/find_ranges_grid_ds.cuh b/IEProtLib/tf_ops/cu/header/find_ranges_grid_ds.cuh
@@ -22,6 +22,7 @@
 #include "defines.hpp"
 #include "gpu_device.hpp"
 #include <memory>
+#include <vector>
 
 namespace mccnn{
 
@@ -76,4 +77,4 @@ namespace mccnn{
         std::vector<int>& pOutVector);
 }
 
-#endif
+#endif