tvm
Loading...
Searching...
No Matches
function.h
Go to the documentation of this file.
1/*
2 * Licensed to the Apache Software Foundation (ASF) under one
3 * or more contributor license agreements. See the NOTICE file
4 * distributed with this work for additional information
5 * regarding copyright ownership. The ASF licenses this file
6 * to you under the Apache License, Version 2.0 (the
7 * "License"); you may not use this file except in compliance
8 * with the License. You may obtain a copy of the License at
9 *
10 * http://www.apache.org/licenses/LICENSE-2.0
11 *
12 * Unless required by applicable law or agreed to in writing,
13 * software distributed under the License is distributed on an
14 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
15 * KIND, either express or implied. See the License for the
16 * specific language governing permissions and limitations
17 * under the License.
18 */
19#ifndef TVM_S_TIR_FUNCTION_H_
20#define TVM_S_TIR_FUNCTION_H_
21
22namespace tvm {
23namespace s_tir {
24namespace attr {
30constexpr const char* kIsScheduled = "tirx.is_scheduled";
31
33constexpr const char* kLayoutFreeTensors = "layout_free_tensors";
40constexpr const char* kLayoutTransforms = "layout_transforms";
41constexpr const char* kEstimatedFlops = "estimated_flops";
42
43} // namespace attr
44} // namespace s_tir
45} // namespace tvm
46#endif // TVM_S_TIR_FUNCTION_H_
constexpr const char * kLayoutFreeTensors
Mark the tensors which is const access and can be transformed layout.
Definition function.h:33
constexpr const char * kEstimatedFlops
Definition function.h:41
constexpr const char * kLayoutTransforms
Marks the layout transforms to be used for a tensor.
Definition function.h:40
constexpr const char * kIsScheduled
Mark the function as scheduled, so the default schedule will pass will skip it.
Definition function.h:30
An object that builds and maintains block scope and StmtSref mapping for Dependence analysis.
Definition attr.h:23