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

Function MakeSequenceGlobalImpl

catboost/cuda/cuda_util/fill.cpp:194–204  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

192
193template <typename T>
194static void MakeSequenceGlobalImpl(
195 TCudaBuffer<T, NCudaLib::TStripeMapping>& buffer,
196 ui32 stream) {
197 auto offset = CreateDistributedObject<T>(0);
198 for (ui32 dev = 0; dev < offset.DeviceCount(); ++dev) {
199 offset.Set(dev, buffer.GetMapping().DeviceSlice(dev).Left);
200 }
201
202 using TKernel = TMakeSequenceKernel<T>;
203 LaunchKernels<TKernel>(buffer.NonEmptyDevices(), stream, buffer, offset);
204}
205
206#define Y_CATBOOST_CUDA_F_IMPL(T) \
207 template <> \

Callers

nothing calls this directly

Calls 5

DeviceCountMethod · 0.80
GetMappingMethod · 0.80
SetMethod · 0.45
DeviceSliceMethod · 0.45
NonEmptyDevicesMethod · 0.45

Tested by

no test coverage detected