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

Function IsCpuOnly

oneflow/core/operator/operator.cpp:1295–1299  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

1293}
1294
1295bool IsCpuOnly(const std::string& user_op_type_name) {
1296 auto* registration_val = user_op::UserOpRegistryMgr::Get().GetOpRegistryResult(user_op_type_name);
1297 CHECK(registration_val != nullptr) << "user_op_type_name: " << user_op_type_name;
1298 return registration_val->cpu_only_supported;
1299}
1300
1301bool IsCpuOnly(const OperatorConf& op_conf) {
1302 OperatorConf::OpTypeCase op_type_case = op_conf.op_type_case();

Callers 7

ApplyImplMethod · 0.85
GetParallelDescMethod · 0.85
CheckAndConstructOpFunction · 0.85
ConstructOpFunction · 0.85

Calls 3

GetFunction · 0.85
GetOpRegistryResultMethod · 0.80
DebugStringMethod · 0.45

Tested by

no test coverage detected