Skip to content

target = None error while tuing an autotvm task #7790

Description

@huochaitiantang

After PR7534(#7534), while tuning an autotvm task, self.target = None, self.target_host = None (default) in class tvm.autotvm.task.Task, will introduce the error :

@zxybazh

Traceback (most recent call last):
  File "test_quantize.py", line 66, in <module>
    tuner.tune(20, logfile=logfile)
  File "~/Desktop/tvms/tvm-oplib/tvmoplib/tvm_tuner/onnx_tuner.py", line 215, in tune
    tuner.load_history(tvm.autotvm.record.load_from_file(tmp_log_file))
  File "~/Desktop/tvms/tvm/python/tvm/autotvm/tuner/model_based_tuner.py", line 294, in load_history
    success = base_model.fit_log(data_set, self.plan_size)
  File "~/Desktop/tvms/tvm/python/tvm/autotvm/tuner/xgboost_cost_model.py", line 250, in fit_log
    res = pool.map(feature_extract_func, data)
  File "~/.pyenv/versions/3.6.4/lib/python3.6/multiprocessing/pool.py", line 266, in map
    return self._map_async(func, iterable, mapstar, chunksize).get()
  File "~/.pyenv/versions/3.6.4/lib/python3.6/multiprocessing/pool.py", line 644, in get
    raise self._value
  File "~/.pyenv/versions/3.6.4/lib/python3.6/multiprocessing/pool.py", line 424, in _handle_tasks
    put(task)
  File "~/.pyenv/versions/3.6.4/lib/python3.6/multiprocessing/connection.py", line 206, in send
    self._send_bytes(_ForkingPickler.dumps(obj))
  File "~/.pyenv/versions/3.6.4/lib/python3.6/multiprocessing/reduction.py", line 51, in dumps
    cls(buf, protocol).dump(obj)
  File "~/Desktop/tvms/tvm/python/tvm/autotvm/task/task.py", line 179, in __getstate__
    self.target, self.target_host
  File "~/Desktop/tvms/tvm/python/tvm/target/target.py", line 197, in check_and_update_host_consist
    target = Target(target, host)
  File "~/Desktop/tvms/tvm/python/tvm/target/target.py", line 98, in __init__
    raise ValueError("target has to be a string or dictionary.")
ValueError: target has to be a string or dictionary.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions