| 1293 | } |
| 1294 | |
| 1295 | bool 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 | |
| 1301 | bool IsCpuOnly(const OperatorConf& op_conf) { |
| 1302 | OperatorConf::OpTypeCase op_type_case = op_conf.op_type_case(); |
no test coverage detected