Conflict with XGBoost when Thrust is enabled

comaniac · November 18, 2020, 11:41pm

I got the same problem and filed a PR to workaround it.

github.com/apache/tvm

Lazy import XGBoost

main ← comaniac:xgb_lazy

opened 11:40PM - 18 Nov 20 UTC

comaniac

+16 -16

I encountered the same issue as this one: https://discuss.tvm.apache.org/t/confl…ict-with-xgboost-when-thrust-is-enabled/6889 with CUDA 10.0 and thrust enabled after upgrading XGBoost to 1.1.0. Since XGBoost is imported by AutoTVM, and AutoTVM is used everywhere (e.g., TOPI, VM, etc), I got the error even with only `from tvm import relay`, as mentioned in the latest reply in the post. Since we cannot reproduce this error in docker and it's a potential issue, this PR makes the XGBoost import lazy. cc @merrymercy @tqchen @vinx13 @zhiics