About Configspace in autotvm

comaniac · September 30, 2020, 8:48pm

This function calculates the total size of a config space:

apache/incubator-tvm/blob/master/python/tvm/autotvm/task/space.py#L838


        # convert schedule axis to space definition axis
        axes = [x if isinstance(x, (VirtualAxis, Axis)) else self.axis(x) for x in axes]

        # add subspace (knob)
        space = space_class(axes, policy, **kwargs)
        self.space_map[name] = space
        self._entity_map[name] = space[0]
        return [Axis(space, i) for i in range(space.num_output)]
    return [Axis(None, i) for i in range(space_class.get_num_output(axes, policy, **kwargs))]

def __len__(self):
    if self._length is None:
        self._length = int(np.prod([len(x) for x in self.space_map.values()]))
    return self._length

def get(self, index):
    """Get a config entity with detailed parameters from this space

    Parameters
    ----------
    index: int

And this function goes through each config in the config space:

github.com

apache/incubator-tvm/blob/master/python/tvm/autotvm/task/space.py#L873


        """get the transform entity(knob) of this entity by name
           do not use this to get a ConfigEntity of this space (should use ConfigSpace.get instead)

        Parameters
        ----------
        name: str
            name of the transform
        """
        return self._entity_map[name]

    def __repr__(self):
        res = "ConfigSpace (len=%d, space_map=\n" % len(self)
        for i, (name, space) in enumerate(self.space_map.items()):
            res += "  %2d %s: %s\n" % (i, name, space)
        return res + ")"


_ann_to_number = {
    "none": 0,
    "vec": 1,
    "unroll": 2,