Features.md missing

yongwww · June 25, 2019, 12:38am

can not find the features.md (mentioned in https://github.com/dmlc/tvm/blob/cffb4fba03ea582417e2630bd163bca773756af6/python/tvm/autotvm/feature.py#L74)

merrymercy · August 2, 2019, 4:48am

You can find some examples and comments

github.com

dmlc/tvm/blob/33ab3c602874d128d600f9cfa5c23da37a9f1ab2/tests/python/unittest/test_autotvm_feature.py#L24


# KIND, either express or implied.  See the License for the
# specific language governing permissions and limitations
# under the License.
"""Test feature extraction"""


import numpy as np


import tvm
from tvm.autotvm import feature


def test_iter_feature_gemm():
N = 128


k = tvm.reduce_axis((0, N), 'k')
A = tvm.placeholder((N, N), name='A')
B = tvm.placeholder((N, N), name='B')
C = tvm.compute(
    A.shape,
    lambda y, x: tvm.sum(A[y, k] * B[k, x], axis=k),
    name='C')

github.com

dmlc/tvm/blob/33ab3c602874d128d600f9cfa5c23da37a9f1ab2/src/autotvm/touch_extractor.cc#L199




void TouchExtractor::ExitMem_() {
}


/*!
* \brief Get axis-based feature for all axes
* \param stmt The statement to be extracted
* \param bool Whether take log for numerical feature
* \param ret_feature The buffer where the return value is stored
*
* \note The format of return value is
* ((
*   ('_itervar_',  var),
*   ('_attr_',     length, nest_level, topdown, bottomup, one_hot_annotation),
*   ('_arith_',    add_ct, mul_ct, div_ct),
*   ('data_vec_0', stride, mod, count, reuse, thread_count, thread_reuse),
*   ('conv_0',     stride, mod, count, reuse, thread_count, thread_reuse),
* ),
* (
*   ('_itervar_',    var2),
*   ('_attr_',       length, nest_level, one_hot_annotation),