fix macro, no !
This commit is contained in:
parent
fa2336fcfd
commit
b71c1b50de
@ -28,9 +28,9 @@ using AFTLoss = xgboost::common::AFTLoss<Distribution>;
|
|||||||
namespace xgboost {
|
namespace xgboost {
|
||||||
namespace obj {
|
namespace obj {
|
||||||
|
|
||||||
#if defined(XGBOOST_USE_CUDA) || !defined(XGBOOST_USE_HIP)
|
#if defined(XGBOOST_USE_CUDA) || defined(XGBOOST_USE_HIP)
|
||||||
DMLC_REGISTRY_FILE_TAG(aft_obj_gpu);
|
DMLC_REGISTRY_FILE_TAG(aft_obj_gpu);
|
||||||
#endif // defined(XGBOOST_USE_CUDA) || !defined(XGBOOST_USE_HIP)
|
#endif // defined(XGBOOST_USE_CUDA) || defined(XGBOOST_USE_HIP)
|
||||||
|
|
||||||
class AFTObj : public ObjFunction {
|
class AFTObj : public ObjFunction {
|
||||||
public:
|
public:
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user