嘉渊
2023-06-19 289cb1d2c8d2fc5a54e9b0fb07b2c33800408d42
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