Sycl implementation for objective functions (#9846)
--------- Co-authored-by: Dmitry Razdoburdin <>
This commit is contained in:
committed by
GitHub
parent
ddab49a8be
commit
43897b8296
28
tests/cpp/plugin/test_sycl_multiclass_obj.cc
Normal file
28
tests/cpp/plugin/test_sycl_multiclass_obj.cc
Normal file
@@ -0,0 +1,28 @@
|
||||
/*!
|
||||
* Copyright 2018-2023 XGBoost contributors
|
||||
*/
|
||||
#include <gtest/gtest.h>
|
||||
#include <xgboost/context.h>
|
||||
|
||||
#include "../objective/test_multiclass_obj.h"
|
||||
|
||||
namespace xgboost {
|
||||
|
||||
TEST(SyclObjective, SoftmaxMultiClassObjGPair) {
|
||||
Context ctx;
|
||||
ctx.UpdateAllowUnknown(Args{{"device", "sycl"}});
|
||||
TestSoftmaxMultiClassObjGPair(&ctx);
|
||||
}
|
||||
|
||||
TEST(SyclObjective, SoftmaxMultiClassBasic) {
|
||||
Context ctx;
|
||||
ctx.UpdateAllowUnknown(Args{{"device", "sycl"}});
|
||||
TestSoftmaxMultiClassObjGPair(&ctx);
|
||||
}
|
||||
|
||||
TEST(SyclObjective, SoftprobMultiClassBasic) {
|
||||
Context ctx;
|
||||
ctx.UpdateAllowUnknown(Args{{"device", "sycl"}});
|
||||
TestSoftprobMultiClassBasic(&ctx);
|
||||
}
|
||||
} // namespace xgboost
|
||||
Reference in New Issue
Block a user