MegEngine/src/custom/impl/utils.cpp

/**
 * \file src/custom/impl/utils.cpp
 * MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
 *
 * Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 */

#include "megbrain/custom/utils.h"
#include "megbrain/common.h"
#include <sstream>

using namespace mgb;

namespace custom {

void assert_failed_log(const char *file, int line, const char *func, const char *expr, const char *msg_fmt, ...) {
    std::string msg = ssprintf("`%s' is true at %s:%d: %s", expr, file, line, func);
    if (msg_fmt) {
        msg_fmt = convert_fmt_str(msg_fmt);
        va_list ap;
        va_start(ap, msg_fmt);
        msg.append("\nextra message: ");
        msg.append(svsprintf(msg_fmt, ap));
        va_end(ap);
    }
    printf("%s\n", msg.c_str());
}

UnImpleWarnLog::UnImpleWarnLog(const std::string &func, const std::string &attr,
                               const std::string &val) {
    mgb_log_warn("you are using the default custom %s function, the `%s` attribute "
                 "of all the outputs tensor will be the same with inputs tensor[0]. "
                 "If there is no input tensor, it will be `%s`", 
                 func.c_str(), attr.c_str(), val.c_str());
}

}