tvm_tirx.backend.cuda.op.timer_start

Contents

tvm_tirx.backend.cuda.op.timer_start#

tvm_tirx.backend.cuda.op.timer_start(event_type, profiler_buffer, profiler_tag, profiler_write_offset, profiler_write_stride, leader_cond, *, ty=None, loc=UNKNOWN_LOC)#

TVM intrinsic for starting the timer for profiling a specific event, and storing profiling result in a buffer.

Parameters:

  • event_type (Enum) – The event to profile.

  • profiler_buffer (Var) – The buffer to store the profiling result.

  • profiler_tag (Var) – Buffer of length 1 storing the base tag of the current thread.

  • profiler_write_offset (Var) – Buffer of length 1 storing the offset in buffer to write the next profiling result for the current thread.

  • profiler_write_stride (int) – The stride to advance in buffer in the next write.

  • leader_cond (Expr) – The condition to check if the current thread is the leader.

  • loc (Location)

Returns:

call (Expr) – The call expression.

Contents