File: test_sycl_prediction_cache.cc

package info (click to toggle)
xgboost 3.0.0-1
  • links: PTS, VCS
  • area: main
  • in suites: trixie
  • size: 13,796 kB
  • sloc: cpp: 67,502; python: 35,503; java: 4,676; ansic: 1,426; sh: 1,320; xml: 1,197; makefile: 204; javascript: 19
file content (23 lines) | stat: -rw-r--r-- 620 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
/**
 * Copyright 2020-2024 by XGBoost contributors
 */
#include <gtest/gtest.h>

#pragma GCC diagnostic push
#pragma GCC diagnostic ignored "-Wtautological-constant-compare"
#pragma GCC diagnostic ignored "-W#pragma-messages"
#include "../tree/test_prediction_cache.h"
#pragma GCC diagnostic pop

namespace xgboost::sycl::tree {

class SyclPredictionCache : public xgboost::TestPredictionCache {};

TEST_F(SyclPredictionCache, Hist) {
  Context ctx;
  ctx.UpdateAllowUnknown(Args{{"device", "sycl"}});

  this->RunTest(&ctx, "grow_quantile_histmaker_sycl", "one_output_per_tree");
}

}  // namespace xgboost::sycl::tree