diff --git a/llvm/projects/hpvm-tensor-rt/tensor_runtime/include/approx_api.h b/llvm/projects/hpvm-tensor-rt/tensor_runtime/include/approx_api.h index ed1c373f6557f08897f6fd16ae7e1b5d17c3c28e..61936f3c2e484c24d2bb57020e6fcf6078a2f773 100644 --- a/llvm/projects/hpvm-tensor-rt/tensor_runtime/include/approx_api.h +++ b/llvm/projects/hpvm-tensor-rt/tensor_runtime/include/approx_api.h @@ -38,6 +38,12 @@ extern "C"{ int conv_mode, int conv_groups, int row, int col, int start); + void* tensorConvPerfSim(void* input_ptr, void* filter_ptr, + int vertical_pad, int horizontal_pad, + int vertical_stride, int horizontal_stride, + int conv_mode, int conv_groups, int row, int col); + + void* tensorConvPerfCudaHalf(void* input_ptr, void* filter_ptr, int vertical_pad, int horizontal_pad, int vertical_stride, int horizontal_stride, diff --git a/llvm/projects/hpvm-tensor-rt/tensor_runtime/include/tensor_runtime.h b/llvm/projects/hpvm-tensor-rt/tensor_runtime/include/tensor_runtime.h index c035050442f776292a09af584a55bfe73a57513a..eb7f85756daf88fde20ffc200fdae6c5a8708fc8 100644 --- a/llvm/projects/hpvm-tensor-rt/tensor_runtime/include/tensor_runtime.h +++ b/llvm/projects/hpvm-tensor-rt/tensor_runtime/include/tensor_runtime.h @@ -234,7 +234,7 @@ extern "C"{ void* quantizeTensorPromise(void* input_ptr, float min, float max); void* addPromiseError(void* x_ptr, int error_scale); void readSkipTensors(int* skip_tensor_ids, int op_count); - + void convertToFP32(struct Tensor* tensor); }