MCPcopy Create free account
hub / github.com/Oneflow-Inc/oneflow / GetDeviceCount

Function GetDeviceCount

oneflow/api/python/symbol/placement_symbol.cpp:40–42  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

38namespace {
39
40int64_t GetDeviceCount(const std::string& device_name) {
41 return Singleton<ep::DeviceManagerRegistry>::Get()->GetDeviceCount(device_name);
42}
43
44struct PlacementSymbolExportUtil {
45 static Maybe<void> CheckDeviceTag(const std::string& type) {

Callers 1

AllDevicePlacementMethod · 0.70

Calls 2

GetFunction · 0.85
GetDeviceCountMethod · 0.45

Tested by

no test coverage detected