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

Method Start

catboost/cuda/cuda_lib/cuda_manager.cpp:200–210  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

198}
199
200void TCudaManager::Start(const NCudaLib::TDeviceRequestConfig& config) {
201 CB_ENSURE(State == nullptr);
202 State.Reset(new TCudaManagerState());
203 CB_ENSURE(!HasDevices());
204 SetDevices(GetDevicesProvider().RequestDevices(config));
205 if (config.EnablePeers) {
206 EnablePeers();
207 State->PeersSupportEnabled = true;
208 }
209 CreateProfiler();
210}
211
212void TCudaManager::Stop() {
213 CB_ENSURE(!IsChildManager);

Callers 11

mode_select_featuresFunction · 0.45
DoMethod · 0.45
ModeFitImplMethod · 0.45
Y_UNIT_TESTFunction · 0.45
StartCudaManagerFunction · 0.45
RequestDeviceMethod · 0.45
mainFunction · 0.45
Y_UNIT_TESTFunction · 0.45

Calls 2

RequestDevicesMethod · 0.80
ResetMethod · 0.45

Tested by

no test coverage detected