stmt.h

Go to the documentation of this file.

/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */
#ifndef TVM_S_TIR_STMT_H_
#define TVM_S_TIR_STMT_H_
 
#include <tvm/ir/prim/expr.h>
 
namespace tvm {
namespace s_tir {
using namespace tvm::prim;
namespace attr {
 
constexpr const char* async_commit_queue_scope = "async_commit_queue_scope";
constexpr const char* async_wait_queue_scope = "async_wait_queue_scope";
constexpr const char* async_wait_inflight_count = "async_wait_inflight_count";
 
constexpr const char* async_scope = "async_scope";
 
constexpr const char* buffer_bound = "buffer_bound";
 
constexpr const char* double_buffer_scope = "double_buffer_scope";
 
constexpr const char* double_buffer_write = "double_buffer_write";
 
constexpr const char* fragment_shape = "fragment_shape";
 
constexpr const char* fragment_layout = "fragment_layout";
 
constexpr const char* pragma_loop_partition_hint = "pragma_loop_partition_hint";
 
constexpr const char* reduce_scope = "reduce_scope";
 
constexpr const char* virtual_thread = "virtual_thread";
 
// -----------------------------------------------------------------------
// meta_schedule annotations
// -----------------------------------------------------------------------
 
constexpr const char* meta_schedule_tiling_structure = "meta_schedule.tiling_structure";
 
constexpr const char* meta_schedule_cooperative_fetch = "meta_schedule.cooperative_fetch";
 
constexpr const char* meta_schedule_thread_extent_low_inclusive =
    "meta_schedule.thread_extent_low_inclusive";
 
constexpr const char* meta_schedule_thread_extent_high_inclusive =
    "meta_schedule.thread_extent_high_inclusive";
 
constexpr const char* meta_schedule_random_compute_producer =
    "meta_schedule.random_compute_producer";
 
constexpr const char* meta_schedule_parallel = "meta_schedule.parallel";
 
constexpr const char* meta_schedule_vectorize = "meta_schedule.vectorize";
 
constexpr const char* meta_schedule_unroll_explicit = "meta_schedule.unroll_explicit";
 
constexpr const char* meta_schedule_unroll_implicit = "meta_schedule.unroll_implicit";
 
constexpr const char* meta_schedule_auto_tensorize = "meta_schedule.auto_tensorize";
 
constexpr const char* meta_schedule_layout_rewrite_preproc = "meta_schedule.layout_rewrite_preproc";
 
constexpr const char* meta_schedule_auto_tensorize_init = "meta_schedule.auto_tensorize_init";
 
constexpr const char* meta_schedule_inline_rule = "meta_schedule.inline_rule";
 
// -----------------------------------------------------------------------
// Schedule primitive / SBlock annotations
// -----------------------------------------------------------------------
 
constexpr const char* script_parsing_detect_access = "tirx.script_parsing_detect_access";
 
constexpr const char* require_block_var_bound_predicate = "require_bound_predicate";
 
constexpr const char* software_pipeline_stage = "software_pipeline_stage";
 
constexpr const char* software_pipeline_order = "software_pipeline_order";
 
constexpr const char* software_pipeline_async_stages = "software_pipeline_async_stages";
 
constexpr const char* layout_free_buffers = "layout_free_buffers";
 
constexpr const char* manifest_shared_memory_local_stage =
    "tirx.manifest_shared_memory_local_stage";
 
constexpr const char* buffer_dim_align = "buffer_dim_align";
 
constexpr const char* explicit_read_region = "explicit_read_region";
 
constexpr const char* explicit_write_region = "explicit_write_region";
 
constexpr const char* irregular_loop_mark = "irregular_loop_mark";
 
constexpr const char* auto_copy = "auto_copy";
 
constexpr const char* local_stage = "local_stage";
 
constexpr const char* vector_bytes = "vector_bytes";
 
constexpr const char* warp_execution = "warp_execution";
 
constexpr const char* layout_transforms = "layout_transforms";
 
constexpr const char* hand_threaded = "hand_threaded";
 
}  // namespace attr
}  // namespace s_tir
}  // namespace tvm
#endif  // TVM_S_TIR_STMT_H_