MCPcopy Create free account
hub / github.com/MegEngine/MegCC / IsCVAvailable

Method IsCVAvailable

compiler/lib/KernelGen/GeneralIntrinsic/CvtColor.cpp:449–454  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

447} // namespace
448
449bool CvtColorKernel::IsCVAvailable(TContext* context) const {
450 auto src_dtype = context->getAttrOprand("operand:0").dtype;
451 bool dtype_ok = Utils::is_int_dtype(src_dtype, 8);
452 bool mode_ok = g_cvt_reg.usable(context->getAttrStr("mode"));
453 return dtype_ok && mode_ok;
454}
455
456//! kernel gen
457std::string CvtColorKernel::GetCVKernelSubSymbol(TContext* context) const {

Callers

nothing calls this directly

Calls 4

is_int_dtypeFunction · 0.85
getAttrOprandMethod · 0.80
getAttrStrMethod · 0.80
usableMethod · 0.45

Tested by

no test coverage detected