tvm
Namespaces | Functions
algorithms.h File Reference

The memory planning algorithm for USMP. More...

#include <tvm/tir/usmp/utils.h>
Include dependency graph for algorithms.h:

Go to the source code of this file.

Namespaces

 tvm
 runtime implementation for LibTorch/TorchScript.
 
 tvm::tir
 
 tvm::tir::usmp
 
 tvm::tir::usmp::algo
 

Functions

Map< BufferInfo, PoolAllocation > tvm::tir::usmp::algo::GreedyBySize (const Array< BufferInfo > &buffer_info_arr, const Integer &memory_pressure)
 The Greedy-by-Size algorithm to plan memory. More...
 
Map< BufferInfo, PoolAllocation > tvm::tir::usmp::algo::GreedyByConflicts (const Array< BufferInfo > &buffer_info_arr, const Integer &memory_pressure)
 The Greedy-by-Conflicts algorithm to plan memory. More...
 
Map< BufferInfo, PoolAllocation > tvm::tir::usmp::algo::HillClimb (const Array< BufferInfo > &buffer_info_arr, const Integer &memory_pressure)
 The Hill-Climb algoritm to plan memory. More...
 

Detailed Description

The memory planning algorithm for USMP.