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 2dc985a0c14ebc18a68d5e54f78bd416f9d3b523..ff0b2a8c68d44b26982423d02ea891136ae87d8e 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);