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

Method IsCVAvailable

compiler/lib/KernelGen/Arm/ArmCommon/CvtColor.cpp:432–437  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

430} // namespace
431
432bool CvtColorKernel::IsCVAvailable(TContext* context) const {
433 auto src_dtype = context->getAttrOprand("operand:0").dtype;
434 bool dtype_ok = Utils::is_int_dtype(src_dtype, 8);
435 bool mode_ok = g_cvt_reg.usable(context->getAttrStr("mode"));
436 return dtype_ok && mode_ok;
437}
438
439//! kernel gen
440std::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