|
| constexpr const char * | kIsScheduled = "tirx.is_scheduled" |
| | Mark the function as scheduled, so the default schedule will pass will skip it.
|
| |
| constexpr const char * | kLayoutFreeTensors = "layout_free_tensors" |
| | Mark the tensors which is const access and can be transformed layout.
|
| |
| constexpr const char * | kLayoutTransforms = "layout_transforms" |
| | Marks the layout transforms to be used for a tensor.
|
| |
| constexpr const char * | kEstimatedFlops = "estimated_flops" |
| |
| constexpr const char * | kThreadBinding = "thread_binding" |
| | Hardware thread axis bound to a parallel loop during S-TIR scheduling.
|
| |
| constexpr const char * | kDoubleBufferScope = "double_buffer_scope" |
| | SBlock annotation selecting a write-tensor index for double buffering in InjectSoftwarePipeline.
|
| |
| constexpr const char * | kFragmentShape = "fragment_shape" |
| | String-valued allocation Call attribute containing the TensorCore fragment shape.
|
| |
| constexpr const char * | kFragmentLayout = "fragment_layout" |
| | String-valued allocation Call attribute containing the TensorCore fragment layout.
|
| |
| constexpr const char * | kLoopPartitionHint = "loop_partition_hint" |
| | For annotation requesting partitioning when its PrimExpr value is provably true.
|
| |
| constexpr const char * | kMetaScheduleTilingStructure = "meta_schedule.tiling_structure" |
| | Mark the tiling structure of blocks that are applied by rule Multi-Level-Tiling.
|
| |
| constexpr const char * | kMetaScheduleCooperativeFetch = "meta_schedule.cooperative_fetch" |
| | Mark that the loop should be further skip and bound to environment threads to enable cooperative fetching.
|
| |
| constexpr const char * | kMetaScheduleThreadExtentLowInclusive |
| | The allowed range of thread extent in thread bindings.
|
| |
| constexpr const char * | kMetaScheduleThreadExtentHighInclusive |
| | The allowed range of thread extent in thread bindings.
|
| |
| constexpr const char * | kMetaScheduleRandomComputeProducer = "meta_schedule.random_compute_producer" |
| | Mark the block whose producer needs to be applied by rule Random-Compute-Location.
|
| |
| constexpr const char * | kMetaScheduleParallel = "meta_schedule.parallel" |
| | Mark auto-parallel setting on the block.
|
| |
| constexpr const char * | kMetaScheduleVectorize = "meta_schedule.vectorize" |
| | Mark auto-vectorize setting on the block.
|
| |
| constexpr const char * | kMetaScheduleUnrollExplicit = "meta_schedule.unroll_explicit" |
| | Mark auto-unroll setting on the block.
|
| |
| constexpr const char * | kMetaScheduleUnrollImplicit = "meta_schedule.unroll_implicit" |
| | Mark auto-unroll setting on the block.
|
| |
| constexpr const char * | kMetaScheduleAutoTensorize = "meta_schedule.auto_tensorize" |
| | Mark that a block should be further rewritten using tensorization.
|
| |
| constexpr const char * | kMetaScheduleLayoutRewritePreproc = "meta_schedule.layout_rewrite_preproc" |
| | Mark that a block is a preprocessor block for layout rewrite.
|
| |
| constexpr const char * | kMetaScheduleAutoTensorizeInit = "meta_schedule.auto_tensorize_init" |
| | Mark that the init statement of a block should be further rewritten using tensorization.
|
| |
| constexpr const char * | kMetaScheduleInlineRule = "meta_schedule.inline_rule" |
| | Mark that a block is disallowed in auto inline.
|
| |
| constexpr const char * | kTensorAllocatedAddr = "s_tir.tensor_allocated_addr" |
| | TensorAllocatedAddresses for block-owned allocations and match tensors.
|
| |
| constexpr const char * | kScriptParsingDetectAccess = "tirx.script_parsing_detect_access" |
| | Mark whether the script-completer need to fill in missing access region during script parsing.
|
| |
| constexpr const char * | kRequireBlockVarBoundPredicate = "require_bound_predicate" |
| | Mark that the block need to add predicate for block var bounds during lowering.
|
| |
| constexpr const char * | kSoftwarePipelineStage = "software_pipeline_stage" |
| | Mark the stage of a statement in the software pipeline.
|
| |
| constexpr const char * | kSoftwarePipelineOrder = "software_pipeline_order" |
| | Mark the order of a statement in the software pipeline.
|
| |
| constexpr const char * | kSoftwarePipelineAsyncStages = "software_pipeline_async_stages" |
| | List stages in the software pipeline that should run asynchronously.
|
| |
| constexpr const char * | kManifestSharedMemoryLocalStage = "tirx.manifest_shared_memory_local_stage" |
| | Mark the local stage for the shared memory access should be added.
|
| |
| constexpr const char * | kTensorDimAlign = "tensor_dim_align" |
| | Mark alignment of tensor dimension The annotation value is an array of explicit tuples (tensor_index, axis, factor, offset). This requires the stride of an axis to be k * factor + offset.
|
| |
| constexpr const char * | kExplicitReadRegion = "explicit_read_region" |
| | Mark that a block has an explicitly specified read region. This is used to override the default read region inference in TIR.
|
| |
| constexpr const char * | kExplicitWriteRegion = "explicit_write_region" |
| | Mark that a block has an explicitly specified write region. This is used to override the default write region inference in TIR.
|
| |
| constexpr const char * | kIrregularLoopMark = "irregular_loop_mark" |
| | ,ark a ForNode represent an irregular loop of non-structural control flow edges.
|
| |
| constexpr const char * | kAutoCopy = "auto_copy" |
| | Mark auto copy for memhammer.
|
| |
| constexpr const char * | kLocalStage = "local_stage" |
| | Mark local stage constraint on data copy.
|
| |
| constexpr const char * | kVectorBytes = "vector_bytes" |
| | Mark vectorization length constraint on block.
|
| |
| constexpr const char * | kWarpExecution = "warp_execution" |
| | Mark that a block is executed by a warp. This implies the extend of threadIdx.x is warp size.
|
| |
| constexpr const char * | kPermutedLayout = "permuted_layout" |
| |
| constexpr const char * | kScheduleRule = "schedule_rule" |
| |
| constexpr const char * | kMetaScheduleWriteCacheLevel = "s_tir.meta_schedule.write_cache_level" |
| |
