tvm_tirx.backend.cuda.op.timer_init

Contents

tvm_tirx.backend.cuda.op.timer_init#

tvm_tirx.backend.cuda.op.timer_init(profiler_buffer, profiler_tag, profiler_write_offset, num_groups, group_id, *, ty=None, loc=UNKNOWN_LOC)#

TVM intrinsic for initializing the CUDA profiler, and store profiling result in a buffer.

Parameters:

  • 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.

  • num_groups (int) – The number of groups in the profiler.

  • group_id (Expr) – The group id of the current thread.

  • loc (Location)

Returns:

call (Expr) – The call expression.

Contents