From 9fb7e98c0b7dcada9eb565cc4ae208d0cf0bf845 Mon Sep 17 00:00:00 2001 From: Rui Xia Date: Tue, 16 Apr 2019 23:58:58 -0700 Subject: [PATCH] fix the allocator type in lru of cuda conv algorithm cache. (#848) --- onnxruntime/core/providers/cuda/nn/conv.h | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/onnxruntime/core/providers/cuda/nn/conv.h b/onnxruntime/core/providers/cuda/nn/conv.h index 0ae0ae5be7..f6fe32531a 100644 --- a/onnxruntime/core/providers/cuda/nn/conv.h +++ b/onnxruntime/core/providers/cuda/nn/conv.h @@ -44,7 +44,6 @@ struct vector_hash { template , typename KeyEqual = std::equal_to, - typename Allocator = std::allocator>, typename ListAllocator = std::allocator> class lru_unordered_map { public: @@ -96,13 +95,14 @@ private: T value; iterator_type lru_iterator; }; + using MapAllocator = std::allocator>; void move_to_front(iterator_type it) { lru_list_.splice(lru_list_.begin(), lru_list_, it); } size_t max_size_; - std::unordered_map items_; + std::unordered_map items_; list_type lru_list_; };