setup

游雁

2023-05-24 d86b4531e0c64494f6a150e8220c7c6e7907d3dc

 funasr/runtime/onnxruntime/src/tensor.h

@@ -5,6 +5,8 @@

using namespace std;

namespace funasr {

template <typename T> class Tensor {
  private:
    void alloc_buff();
@@ -71,7 +73,7 @@
{
    buff_size = size[0] * size[1] * size[2] * size[3];
    mem_size = buff_size;
    buff = (T *)aligned_malloc(32, buff_size * sizeof(T));
    buff = (T *)AlignedMalloc(32, buff_size * sizeof(T));
}

template <typename T> void Tensor<T>::free_buff()
@@ -152,4 +154,6 @@
    fwrite(buff, 1, buff_size * sizeof(T), fp);
    fclose(fp);
}

} // namespace funasr
#endif