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

Method NonEmptyDevices

catboost/cuda/cuda_lib/mapping.h:63–72  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

61 }
62
63 TDevicesList NonEmptyDevices() const {
64 TDevicesListBuilder builder;
65 auto devCount = GetCudaManager().GetDeviceCount();
66 for (ui64 dev = 0; dev < devCount; ++dev) {
67 if (MemoryUsageAt(dev)) {
68 builder.AddDevice(dev);
69 }
70 }
71 return builder.Build();
72 }
73
74 template <class TFunc>
75 inline void Apply(TFunc&& trans) const {

Callers 15

GetSrcSlicesByDevicesMethod · 0.45
RunMethod · 0.45
WriteMethod · 0.45
SubmitReadAsyncMethod · 0.45
Y_UNIT_TESTFunction · 0.45
SegmentedScanVectorImplFunction · 0.45
ReorderOneBitImplFunction · 0.45
NonZeroFilterImplFunction · 0.45
DecompressImplFunction · 0.45
CompressImplFunction · 0.45
GatherFromCompressedImplFunction · 0.45

Calls 2

GetDeviceCountMethod · 0.45
BuildMethod · 0.45

Tested by 4

Y_UNIT_TESTFunction · 0.36
Y_UNIT_TESTFunction · 0.36
Y_UNIT_TESTFunction · 0.36
Y_UNIT_TESTFunction · 0.36