Hello!

After debugging I have some new discoveries

In the 
https://github.com/apache/incubator-tvm/blob/1014fefa54b5f0a359501b6d19ea3b5a52d6dca6/python/tvm/autotvm/measure/local_executor.py#L130

Parameter "do_fork" will affect the success of autotvm when the schedule 
increases the acquisition of CUDA version 

I add "self.do_fork=False" in line 145:


    def submit(self, func, *args, **kwargs):
        self.do_fork = False
        if not self.do_fork:
            return LocalFutureNoFork(func(*args, **kwargs))

        queue = Queue(2)
        process = Process(target=call_with_timeout,
                          args=(queue, self.timeout, func, args, kwargs))
        process.start()
        return LocalFuture(process, queue)


Through the above modification, autotvm can run normally.

Can I directly control this parameter to run autotvm? Just like the following 
settings:


tuning_option = {

    'log_filename': log_file,
    'tuner': 'ga',
    'n_trial': 2000,
    'early_stopping': 600,
    'measure_option': autotvm.measure_option(
        builder=autotvm.LocalBuilder(timeout=10),
        runner=autotvm.LocalRunner(number=20, repeat=3, timeout=4, 
min_repeat_ms=150)
    ),
}

I will be very grateful if you can help me.





---
[Visit 
Topic](https://discuss.tvm.ai/t/autotvm-error-cudaerrorcudartunloading-initialization-error-error-no-2/6487/4)
 to respond.

You are receiving this because you enabled mailing list mode.

To unsubscribe from these emails, [click 
here](https://discuss.tvm.ai/email/unsubscribe/d4f5906023b70a926d295def3701a3fca5dbff4e7b69f87e3b127d13ae9ba2c6).

Reply via email to