MCPcopy Create free account
hub / github.com/catboost/catboost / SegmentedScanVectorImpl

Function SegmentedScanVectorImpl

catboost/cuda/cuda_util/segmented_scan.cpp:71–80  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

69
70template <typename T, typename TMapping, typename TUi32>
71static void SegmentedScanVectorImpl(
72 const TCudaBuffer<T, TMapping>& input,
73 const TCudaBuffer<TUi32, TMapping>& flags,
74 TCudaBuffer<std::remove_const_t<T>, TMapping>& output,
75 bool inclusive,
76 ui32 flagMask,
77 ui32 streamId) {
78 using TKernel = TSegmentedScanKernel<std::remove_const_t<T>>;
79 LaunchKernels<TKernel>(input.NonEmptyDevices(), streamId, input, flags, flagMask, output, inclusive);
80}
81
82#define Y_CATBOOST_CUDA_F_IMPL_PROXY(x) \
83 Y_CATBOOST_CUDA_F_IMPL x

Callers

nothing calls this directly

Calls 1

NonEmptyDevicesMethod · 0.45

Tested by

no test coverage detected