tvm
Classes | Namespaces | Functions
profiling.h File Reference

Runtime profiling including timers. More...

#include <tvm/runtime/c_runtime_api.h>
#include <tvm/runtime/container/map.h>
#include <tvm/runtime/device_api.h>
#include <tvm/runtime/object.h>
#include <tvm/runtime/packed_func.h>
#include <tvm/runtime/registry.h>
#include <stack>
#include <string>
#include <unordered_map>
#include <utility>
#include <vector>
Include dependency graph for profiling.h:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes

class  tvm::runtime::TimerNode
 Base class for all implementations. More...
 
class  tvm::runtime::Timer
 Timer for a specific device. More...
 
struct  tvm::runtime::profiling::DeviceWrapperNode
 Wrapper for Device because Device is not passable across the PackedFunc interface. More...
 
class  tvm::runtime::profiling::DeviceWrapper
 Wrapper for Device. More...
 
class  tvm::runtime::profiling::ReportNode
 Data collected from a profiling run. Includes per-call metrics and per-device metrics. More...
 
class  tvm::runtime::profiling::Report
 
class  tvm::runtime::profiling::MetricCollectorNode
 Interface for user defined profiling metric collection. More...
 
class  tvm::runtime::profiling::MetricCollector
 Wrapper for MetricCollectorNode. More...
 
struct  tvm::runtime::profiling::CallFrame
 
class  tvm::runtime::profiling::Profiler
 
class  tvm::runtime::profiling::DurationNode
 
class  tvm::runtime::profiling::PercentNode
 
class  tvm::runtime::profiling::CountNode
 
class  tvm::runtime::profiling::RatioNode
 

Namespaces

 tvm
 runtime implementation for LibTorch/TorchScript.
 
 tvm::runtime
 
 tvm::runtime::profiling
 

Functions

Timer tvm::runtime::DefaultTimer (Device dev)
 Default timer if one does not exist for the device. More...
 
String tvm::runtime::profiling::ShapeString (const std::vector< NDArray > &shapes)
 String representation of an array of NDArray shapes. More...
 
String tvm::runtime::profiling::ShapeString (NDArray shape, DLDataType dtype)
 String representation of shape encoded as an NDArray. More...
 
String tvm::runtime::profiling::ShapeString (const std::vector< int64_t > &shape, DLDataType dtype)
 String representation of a shape encoded as a vector. More...
 
PackedFunc tvm::runtime::profiling::ProfileFunction (Module mod, std::string func_name, int device_type, int device_id, int warmup_iters, Array< MetricCollector > collectors)
 Collect performance information of a function execution. Usually used with a compiled PrimFunc (via tvm.build). More...
 
PackedFunc tvm::runtime::profiling::WrapTimeEvaluator (PackedFunc f, Device dev, int number, int repeat, int min_repeat_ms, int limit_zero_time_iterations, int cooldown_interval_ms, int repeats_to_cooldown, int cache_flush_bytes=0, PackedFunc f_preproc=nullptr)
 Wrap a timer function to measure the time cost of a given packed function. More...
 

Detailed Description

Runtime profiling including timers.