3d8fd2a2e7c39216482275d04e71aa131216e4c2,tests/python/unittest/test_auto_scheduler_search_policy.py,,test_sketch_search_policy_xgbmodel,#,164

Before Change

def test_sketch_search_policy_xgbmodel():
    // wrap the search in a new thread to avoid the conflict
    // between python"s multiprocessing and tvm"s thread pool
    t = PropagatingThread(
        target=search_common,
        kwargs={
            "cost_model": auto_scheduler.XGBModel(),
        },
    )
    t.start()
    t.join()


@tvm.testing.requires_cuda
def test_sketch_search_policy_cuda_rpc_runner():

After Change

@tvm.testing.requires_llvm
def test_sketch_search_policy_xgbmodel():
    search_common(cost_model=auto_scheduler.XGBModel())


@tvm.testing.requires_cuda
def test_sketch_search_policy_cuda_rpc_runner():

Instances