Home
last modified time | relevance | path

Searched refs:GetDeviceAccessor (Results 1 – 18 of 18) sorted by relevance

/dports/misc/py-xgboost/xgboost-1.5.1/tests/cpp/tree/gpu_hist/
H A Dtest_histogram.cu36 BuildGradientHistogram(page->GetDeviceAccessor(0), in TestDeterministicHistogram()
50 BuildGradientHistogram(page->GetDeviceAccessor(0), in TestDeterministicHistogram()
73 BuildGradientHistogram(page->GetDeviceAccessor(0), in TestDeterministicHistogram()
130 BuildGradientHistogram(page->GetDeviceAccessor(0), in TestGPUHistogramCategorical()
143 BuildGradientHistogram(page->GetDeviceAccessor(0), in TestGPUHistogramCategorical()
/dports/misc/xgboost/xgboost-1.5.1/tests/cpp/tree/gpu_hist/
H A Dtest_histogram.cu36 BuildGradientHistogram(page->GetDeviceAccessor(0), in TestDeterministicHistogram()
50 BuildGradientHistogram(page->GetDeviceAccessor(0), in TestDeterministicHistogram()
73 BuildGradientHistogram(page->GetDeviceAccessor(0), in TestDeterministicHistogram()
130 BuildGradientHistogram(page->GetDeviceAccessor(0), in TestGPUHistogramCategorical()
143 BuildGradientHistogram(page->GetDeviceAccessor(0), in TestGPUHistogramCategorical()
/dports/misc/py-xgboost/xgboost-1.5.1/tests/cpp/data/
H A Dtest_iterative_device_dmatrix.cu29 auto from_iter = page_concatenated->GetDeviceAccessor(0); in TestEquivalent()
39 auto from_data = ellpack.Impl()->GetDeviceAccessor(0); in TestEquivalent()
149 EXPECT_EQ(iterator[1], impl->GetDeviceAccessor(0).NullValue()); in TEST()
150 EXPECT_EQ(iterator[5], impl->GetDeviceAccessor(0).NullValue()); in TEST()
152 EXPECT_EQ(iterator[7], impl->GetDeviceAccessor(0).NullValue()); in TEST()
H A Dtest_ellpack_page.cu92 auto accessor = ellpack.Impl()->GetDeviceAccessor(0); in TEST()
169 … dh::LaunchN(kCols, ReadRowFunction(impl->GetDeviceAccessor(0), current_row, row_d.data().get())); in TEST()
172 …dh::LaunchN(kCols, ReadRowFunction(result.GetDeviceAccessor(0), current_row, row_result_d.data().g… in TEST()
224 dh::LaunchN(kCols, ReadRowFunction(impl->GetDeviceAccessor(0), in TEST()
230 ReadRowFunction(result.GetDeviceAccessor(0), compacted_row, in TEST()
H A Dtest_sparse_page_dmatrix.cu187 … dh::LaunchN(kCols, ReadRowFunction(impl->GetDeviceAccessor(0), current_row, row_d.data().get())); in TEST()
190 …dh::LaunchN(kCols, ReadRowFunction(impl_ext->GetDeviceAccessor(0), current_row, row_ext_d.data().g… in TEST()
/dports/misc/xgboost/xgboost-1.5.1/tests/cpp/data/
H A Dtest_iterative_device_dmatrix.cu29 auto from_iter = page_concatenated->GetDeviceAccessor(0); in TestEquivalent()
39 auto from_data = ellpack.Impl()->GetDeviceAccessor(0); in TestEquivalent()
149 EXPECT_EQ(iterator[1], impl->GetDeviceAccessor(0).NullValue()); in TEST()
150 EXPECT_EQ(iterator[5], impl->GetDeviceAccessor(0).NullValue()); in TEST()
152 EXPECT_EQ(iterator[7], impl->GetDeviceAccessor(0).NullValue()); in TEST()
H A Dtest_ellpack_page.cu92 auto accessor = ellpack.Impl()->GetDeviceAccessor(0); in TEST()
169 … dh::LaunchN(kCols, ReadRowFunction(impl->GetDeviceAccessor(0), current_row, row_d.data().get())); in TEST()
172 …dh::LaunchN(kCols, ReadRowFunction(result.GetDeviceAccessor(0), current_row, row_result_d.data().g… in TEST()
224 dh::LaunchN(kCols, ReadRowFunction(impl->GetDeviceAccessor(0), in TEST()
230 ReadRowFunction(result.GetDeviceAccessor(0), compacted_row, in TEST()
H A Dtest_sparse_page_dmatrix.cu187 … dh::LaunchN(kCols, ReadRowFunction(impl->GetDeviceAccessor(0), current_row, row_d.data().get())); in TEST()
190 …dh::LaunchN(kCols, ReadRowFunction(impl_ext->GetDeviceAccessor(0), current_row, row_ext_d.data().g… in TEST()
/dports/misc/py-xgboost/xgboost-1.5.1/src/data/
H A Dellpack_page.cu220 auto device_accessor = dst->GetDeviceAccessor(device_idx); in CopyDataToEllpack()
250 auto device_accessor = dst->GetDeviceAccessor(device_idx); in WriteNullValues()
450 auto device_accessor = GetDeviceAccessor(device); in CreateHistIndices()
474 EllpackDeviceAccessor EllpackPageImpl::GetDeviceAccessor( in GetDeviceAccessor() function in xgboost::EllpackPageImpl
H A Dellpack_page.cuh198 GetDeviceAccessor(int device,
/dports/misc/xgboost/xgboost-1.5.1/src/data/
H A Dellpack_page.cu220 auto device_accessor = dst->GetDeviceAccessor(device_idx); in CopyDataToEllpack()
250 auto device_accessor = dst->GetDeviceAccessor(device_idx); in WriteNullValues()
450 auto device_accessor = GetDeviceAccessor(device); in CreateHistIndices()
474 EllpackDeviceAccessor EllpackPageImpl::GetDeviceAccessor( in GetDeviceAccessor() function in xgboost::EllpackPageImpl
H A Dellpack_page.cuh198 GetDeviceAccessor(int device,
/dports/misc/py-xgboost/xgboost-1.5.1/src/tree/
H A Dupdater_gpu_hist.cu276 auto matrix = page->GetDeviceAccessor(device_id); in EvaluateRootSplit()
313 auto matrix = page->GetDeviceAccessor(device_id); in EvaluateLeftRightSplits()
366 BuildGradientHistogram(page->GetDeviceAccessor(device_id), in BuildHist()
394 auto d_matrix = page->GetDeviceAccessor(device_id); in UpdatePosition()
468 auto d_matrix = page->GetDeviceAccessor(device_id); in FinalisePositionInPage()
/dports/misc/xgboost/xgboost-1.5.1/src/tree/
H A Dupdater_gpu_hist.cu276 auto matrix = page->GetDeviceAccessor(device_id); in EvaluateRootSplit()
313 auto matrix = page->GetDeviceAccessor(device_id); in EvaluateLeftRightSplits()
366 BuildGradientHistogram(page->GetDeviceAccessor(device_id), in BuildHist()
394 auto d_matrix = page->GetDeviceAccessor(device_id); in UpdatePosition()
468 auto d_matrix = page->GetDeviceAccessor(device_id); in FinalisePositionInPage()
/dports/misc/py-xgboost/xgboost-1.5.1/src/predictor/
H A Dgpu_predictor.cu715 page.Impl()->GetDeviceAccessor(generic_param_->gpu_id, feature_types), in DevicePredictInternal()
1006 EllpackDeviceAccessor data{batch.Impl()->GetDeviceAccessor(generic_param_->gpu_id)}; in PredictLeaf()
/dports/misc/xgboost/xgboost-1.5.1/src/predictor/
H A Dgpu_predictor.cu715 page.Impl()->GetDeviceAccessor(generic_param_->gpu_id, feature_types), in DevicePredictInternal()
1006 EllpackDeviceAccessor data{batch.Impl()->GetDeviceAccessor(generic_param_->gpu_id)}; in PredictLeaf()
/dports/misc/py-xgboost/xgboost-1.5.1/tests/cpp/tree/
H A Dtest_gpu_hist.cu103 page->GetDeviceAccessor(0), maker.feature_groups->DeviceAccessor(0), in TestBuildHist()
/dports/misc/xgboost/xgboost-1.5.1/tests/cpp/tree/
H A Dtest_gpu_hist.cu103 page->GetDeviceAccessor(0), maker.feature_groups->DeviceAccessor(0), in TestBuildHist()