leowang1225 commented on a change in pull request #7185:
URL: https://github.com/apache/tvm/pull/7185#discussion_r551140443
##########
File path: python/tvm/auto_scheduler/measure.py
##########
@@ -303,12 +312,26 @@ class LocalBuilder(ProgramBuilder):
This is used in a wrapper of the multiprocessing.Process.join().
n_parallel : int = multiprocessing.cpu_count()
Number of threads used to build in parallel.
- build_func : str = 'default'
- The name of registered build function.
+ build_func: callable or str = "default"
+ If is 'default', use default build function
+ If is 'ndk', use function for android ndk
+ If is callable, use it as custom build function, expect lib_format
field.
"""
def __init__(self, timeout=15, n_parallel=multiprocessing.cpu_count(),
build_func="default"):
- self.__init_handle_by_constructor__(_ffi_api.LocalBuilder, timeout,
n_parallel, build_func)
+ if build_func == "default":
Review comment:
when auto_scheduler.local_builder.build is called, the context can't get
LocalBuilder class instance
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
[email protected]
