This path checks requested memory is weight gradient which information
will be used for planning.
Signed-off-by: Jiho Chu <jiho.chu@samsung.com>
unsigned int CachePool::requestMemory(size_t bytes, unsigned int start_time,
unsigned int end_time,
std::vector<unsigned int> exec_order,
- TensorLifespan lifespan) {
+ TensorLifespan lifespan, bool is_wgrad) {
auto id = MemoryPool::requestMemory(bytes, start_time, end_time, exec_order,
- lifespan);
+ lifespan, is_wgrad);
const CachePolicy policy = convertTensorLifespanToCachePolicy(lifespan);
virtual unsigned int requestMemory(
size_t bytes, unsigned int start_time, unsigned int end_time,
std::vector<unsigned int> exec_order = std::vector<unsigned int>(),
- TensorLifespan lifespan = TensorLifespan::MAX_LIFESPAN);
+ TensorLifespan lifespan = TensorLifespan::MAX_LIFESPAN,
+ bool is_wgrad = false);
/**
* @brief Get the allocated cache