MCPcopy Create free account
hub / github.com/MegEngine/MegEngine / get

Method get

dnn/src/rocm/convolution/helper.cpp:38–48  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

36
37template <typename Args, typename ValueType>
38std::pair<bool, ValueType> MIOpenCache<Args, ValueType>::get(const Args& args) {
39 std::string key = args.to_miopen_algo_cache_key().to_string_binary();
40 std::lock_guard<std::mutex> guard{m_mtx};
41 auto search = m_cache.find(key);
42 bool find = search != m_cache.end();
43 ValueType val = ValueType();
44 if (find) {
45 val = search->second;
46 }
47 return std::make_pair(find, val);
48}
49
50#define INST(_opr, _miopen_algo) \
51 template class megdnn::rocm::convolution::MIOpenCache< \

Callers 15

getROCMContextMethod · 0.45
exec_internalMethod · 0.45
is_availableMethod · 0.45
find_best_algoMethod · 0.45
exec_internalMethod · 0.45
is_availableMethod · 0.45
find_best_algoMethod · 0.45
exec_internalMethod · 0.45
is_availableMethod · 0.45

Calls 3

to_string_binaryMethod · 0.80
findMethod · 0.45
endMethod · 0.45

Tested by

no test coverage detected