Skip to content

Commit 5a611af

Browse files
author
Pei Yang
authored
fix trt calib not working bug, test=develop (#21934) (#22110)
1 parent 34ef38c commit 5a611af

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

paddle/fluid/framework/ir/graph_pattern_detector.h

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,8 +18,10 @@
1818
#include <gtest/gtest_prod.h>
1919
#endif
2020

21+
#include <map>
2122
#include <memory>
2223
#include <numeric>
24+
#include <set>
2325
#include <string>
2426
#include <unordered_map>
2527
#include <unordered_set>
@@ -294,7 +296,7 @@ class GraphPatternDetector {
294296
using hit_rcd_t =
295297
std::pair<Node* /*node in graph*/, PDNode* /*node in pattern*/>;
296298
PDPattern pattern_;
297-
std::unordered_map<const PDNode*, std::unordered_set<Node*>> pdnodes2nodes_;
299+
std::map<const PDNode*, std::set<Node*>> pdnodes2nodes_;
298300
};
299301

300302
// some helper methods.

0 commit comments

Comments
 (0)