saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-25 22:26:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix the allocator type in lru of cuda conv algorithm cache. (#848)

Browse source
This commit is contained in:
Rui Xia 2019-04-16 23:58:58 -07:00 committed by KeDengMS
parent 41dc3130f5
commit 9fb7e98c0b
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
onnxruntime/core/providers/cuda/nn/conv.h
View file

@ -44,7 +44,6 @@ struct vector_hash {
template <typename Key, typename T,
typename Hash = std::hash<Key>,
typename KeyEqual = std::equal_to<Key>,
typename Allocator = std::allocator<std::pair<const Key, T>>,
typename ListAllocator = std::allocator<Key>>
class lru_unordered_map {
public:
@ -96,13 +95,14 @@ private:
T value;
iterator_type lru_iterator;
};
using MapAllocator = std::allocator<std::pair<const Key, value_type>>;
void move_to_front(iterator_type it) {
lru_list_.splice(lru_list_.begin(), lru_list_, it);
}
size_t max_size_;
std::unordered_map<Key, value_type, Hash, KeyEqual, Allocator> items_;
std::unordered_map<Key, value_type, Hash, KeyEqual, MapAllocator> items_;
list_type lru_list_;
};