utils.h

#ifndef UTILS_H
#define UTILS_H

#include <string>
#include <acl/acl.h>
#include <atb/atb_infer.h>
#include <atb/types.h>
#include <atb/utils.h>
#include "atb/infer_op_params.h"
#include "utils/log.h"

#define CHECK_RET(cond, str) \
    do                       \
    {                        \
        if (cond)            \
        {                    \
            LOG_ERROR(str);  \
            exit(0);         \
        }                    \
    } while (0)

// Set the attributes of each input tensor.
void CreateInTensorDescs(atb::SVector<atb::TensorDesc> &intensorDescs);

// Set each input tensor and allocate memory space for each input tensor. The input tensor is manually set. In the project implementation, you can use torchTensor conversion or other simple data structure conversion.
void CreateInTensors(atb::SVector<atb::Tensor> &inTensors, atb::SVector<atb::TensorDesc> &intensorDescs);

// Set each output tensor and allocate memory space for each output tensor. The setting is the same as that of an input tensor.
void CreateOutTensors(atb::SVector<atb::Tensor> &outTensors, atb::SVector<atb::TensorDesc> &outtensorDescs);

void CreateTensorFromDesc(atb::Tensor &tensor, atb::TensorDesc &tensorDescs);

// Print the output.
void PrintOutTensorValue(atb::Tensor &outTensor);

// Create a graph operator.
atb::Status CreateGraphOperation(atb::Operation **operation);

#endif