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

Method Init

tensorflow/stream_executor/rocm/rocm_gpu_executor.cc:155–176  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

153}
154
155port::Status GpuExecutor::Init(int device_ordinal,
156 DeviceOptions device_options) {
157 device_ordinal_ = device_ordinal;
158
159 auto status = GpuDriver::Init();
160 if (!status.ok()) {
161 return status;
162 }
163
164 status = GpuDriver::GetDevice(device_ordinal_, &device_);
165 if (!status.ok()) {
166 return status;
167 }
168
169 status = GpuDriver::CreateContext(device_ordinal_, device_, device_options,
170 &context_);
171 if (!status.ok()) {
172 return status;
173 }
174
175 return GpuDriver::GetGpuISAVersion(&version_, device_);
176}
177
178bool GpuExecutor::FindOnDiskForComputeCapability(
179 absl::string_view filename, absl::string_view canonical_suffix,

Callers 3

AllocateEventMethod · 0.45
AllocateStreamMethod · 0.45
AllocateTimerMethod · 0.45

Calls 4

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

Tested by

no test coverage detected