MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / Init

Method Init

tensorflow/stream_executor/cuda/cuda_gpu_executor.cc:130–151  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

128}
129
130port::Status GpuExecutor::Init(int device_ordinal,
131 DeviceOptions device_options) {
132 device_ordinal_ = device_ordinal;
133
134 auto status = GpuDriver::Init();
135 if (!status.ok()) {
136 return status;
137 }
138
139 status = GpuDriver::GetDevice(device_ordinal_, &device_);
140 if (!status.ok()) {
141 return status;
142 }
143
144 status = GpuDriver::CreateContext(device_ordinal_, device_, device_options,
145 &context_);
146 if (!status.ok()) {
147 return status;
148 }
149
150 return GpuDriver::GetComputeCapability(&cc_major_, &cc_minor_, device_);
151}
152
153bool GpuExecutor::FindOnDiskForComputeCapability(
154 absl::string_view filename, absl::string_view canonical_suffix,

Callers 3

AllocateEventMethod · 0.45
AllocateStreamMethod · 0.45
AllocateTimerMethod · 0.45

Calls 5

GetDeviceFunction · 0.85
CreateContextFunction · 0.85
InitFunction · 0.50
GetComputeCapabilityFunction · 0.50
okMethod · 0.45

Tested by

no test coverage detected