From 467731fb2d992fa6456904b70552d7f767cc1f9d Mon Sep 17 00:00:00 2001 From: Hashim Sharif <hsharif3@tyler.cs.illinois.edu> Date: Sat, 2 Nov 2019 16:26:00 -0500 Subject: [PATCH] Adding declaration for tensorConvPerfCudaHalf --- .../hpvm-tensor-rt/tensor_runtime/include/approx_api.h | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) 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 2dc985a0c1..ff0b2a8c68 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 @@ -31,8 +31,14 @@ extern "C"{ void* tensorConvPerfCuda(void* input, void* filter, int vertical_pad, int horizontal_pad, int vertical_stride, int horizontal_stride, - int conv_mode, int conv_groups, int row, int col, int start); - + int conv_mode, int conv_groups, + int row, int col, int start); + + void* tensorConvPerfCudaHalf(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, int start); void sampleFilter(Tensor* filter, int skip_rate, int skip_offset); -- GitLab