MCPcopy Create free account
hub / github.com/PaddlePaddle/Paddle / BackendToString

Function BackendToString

paddle/phi/common/backend.h:162–196  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

160}
161
162inline std::string BackendToString(const Backend& backend) {
163 switch (backend) {
164 case Backend::UNDEFINED:
165 return "Undefined(ALL_BACKEND)";
166 case Backend::CPU:
167 return "CPU";
168 case Backend::GPU:
169 return "GPU";
170 case Backend::XPU:
171 return "XPU";
172 case Backend::ONEDNN:
173 return "ONEDNN";
174 case Backend::GPUDNN:
175 return "GPUDNN";
176 case Backend::KPS:
177 return "KPS";
178 case Backend::IPU:
179 return "IPU";
180 case Backend::CUSTOM:
181 return "CUSTOM";
182 default: {
183 size_t device_type_id_ = static_cast<size_t>(backend) -
184 static_cast<size_t>(Backend::NUM_BACKENDS);
185 std::string device_type =
186 phi::CustomRegisteredDeviceMap::Instance().GetGlobalDeviceType(
187 device_type_id_);
188 if (!device_type.empty()) {
189 return device_type;
190 } else {
191 PD_THROW(
192 "Invalid enum backend type `", static_cast<int>(backend), "`.");
193 }
194 }
195 }
196}
197
198inline Backend get_accelerat_backend() {
199#if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP)

Callers 3

RewritePdOpMethod · 0.85

Calls 3

InstanceFunction · 0.85
GetGlobalDeviceTypeMethod · 0.80
emptyMethod · 0.45

Tested by

no test coverage detected