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
