tvm
|
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>
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... | |
Runtime profiling including timers.