mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

fix bug

This commit is contained in:
yefeng 2022-03-15 17:19:00 +08:00
parent 9324b085f2
commit f6823d6485
2 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
mindspore/lite/src/runtime/allocator.cc
View File

@ -16,5 +16,10 @@
#include "src/runtime/inner_allocator.h"
namespace mindspore {
std::shared_ptr<Allocator> Allocator::Create() { return std::make_shared<DefaultAllocator>(); }
std::shared_ptr<Allocator> Allocator::Create() {
#ifdef SERVER_INFERENCE
return nullptr;
#endif
return std::make_shared<DefaultAllocator>();
}
} // namespace mindspore

2
mindspore/lite/src/runtime/numa_adapter.h
View File

@ -57,7 +57,7 @@ class NUMAAdapter {
NUMAAdapter();
~NUMAAdapter();
inline bool Available() const { return false; }
inline bool Available() const { return available_; }
void Bind(int node_id);
void *Malloc(int node_id, size_t size);
void Free(void *data, size_t size);