roll back to unordered map

Oneflow-Inc · Jul 3, 2021 · 409da78 · 409da78
1 parent 22c0540
commit 409da78
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/oneflow/core/framework/random_generator.h b/oneflow/core/framework/random_generator.h
@@ -20,6 +20,7 @@ limitations under the License.
 #include "oneflow/core/device/device_context.h"
 #include <fcntl.h>
 #include <unistd.h>
+#include <unordered_map>
 
 #ifdef WITH_CUDA
 #include <curand.h>
@@ -117,7 +118,7 @@ class AutoGeneratorImpl : public GeneratorImpl {
   }
 
  private:
-  std::map<DeviceType, std::shared_ptr<GeneratorImpl>> generators_;
+  std::unordered_map<DeviceType, std::shared_ptr<GeneratorImpl>, std::hash<int>> generators_;
 };
 
 class Generator final {