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

Method DeviceSlice

catboost/cuda/cuda_lib/mapping.h:134–139  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

132 }
133
134 TSlice DeviceSlice(ui64 dev) const {
135 if (dev != DeviceId) {
136 return {0, 0};
137 }
138 return {0, Count};
139 }
140
141 TSingleMapping RepeatOnAllDevices(ui64 objectCount, ui64 objectSize = 1) const {
142 return TSingleMapping(DeviceId, objectCount, objectSize);

Callers 15

ThroughHostBroadcastFunction · 0.45
DeviceMemoryOffsetMethod · 0.45
SliceViewMethod · 0.45
ColumnsViewMethod · 0.45
DeviceViewMethod · 0.45
ConstDeviceViewMethod · 0.45
CopyMethod · 0.45
StripeViewFunction · 0.45
ParallelStripeViewFunction · 0.45
PassTasksMethod · 0.45
PassTasksMethod · 0.45

Calls

no outgoing calls

Tested by 15

Y_UNIT_TESTFunction · 0.36
Y_UNIT_TESTFunction · 0.36
Y_UNIT_TESTFunction · 0.36
Y_UNIT_TESTFunction · 0.36
RunReorderTestFunction · 0.36
Y_UNIT_TESTFunction · 0.36
CheckDataSetFunction · 0.36
MakeOffsetsBiasFunction · 0.36
SortFunction · 0.36
TestMakeFlagsImplFunction · 0.36