Skip to content

Commit e9cdf46

Browse files
malfetcsarofeen
authored andcommitted
Fix unordered-map-over-enum for GCC 5.4 (pytorch#41063)
Summary: Forgot to add this to pytorch#41055 Pull Request resolved: pytorch#41063 Differential Revision: D22407451 Pulled By: malfet fbshipit-source-id: 6f06653b165cc4817d134657f87caf643182832a
1 parent 48dd16d commit e9cdf46

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torch/csrc/jit/codegen/cuda/lower_thread_predicate.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@ namespace torch {
88
namespace jit {
99
namespace fuser {
1010

11-
const static std::unordered_map<ParallelType, int> pt_to_offset{
11+
const static std::unordered_map<ParallelType, int, TypeHash> pt_to_offset{
1212
{ParallelType::BIDx, 0},
1313
{ParallelType::BIDy, 1},
1414
{ParallelType::BIDz, 2},

0 commit comments

Comments
 (0)