Method exec

dnn/src/common/multi_head_attn/proxy_forward_base.cpp:391–399

}

void MHAForwardProxyBase::exec(MHA_PROXY_FORWARD_EXEC_PARAM) {
#define cb(DType)                                                   \
    if (queries.layout.dtype.enumv() == DTypeTrait<DType>::enumv) { \
        using ctype = typename DTypeTrait<DType>::ctype;            \
        exec_internal<ctype>(MHA_PROXY_FORWARD_CALL, workspace);    \
    }
    MEGDNN_FOREACH_COMPUTING_DTYPE(cb)
#undef cb
}

template <typename T>
void MHAForwardProxyBase::exec_internal(MHA_PROXY_FORWARD_EXEC_PARAM) {