diff --git a/llvm/test/VISC/DNN_Benchmarks/common/include/visc.h b/llvm/test/VISC/DNN_Benchmarks/common/include/visc.h
index ab4787ffa221e225e354e3fed470dbe632a090c4..d4ebbdd4b1327c50ce763debcc37ccac33756929 100644
--- a/llvm/test/VISC/DNN_Benchmarks/common/include/visc.h
+++ b/llvm/test/VISC/DNN_Benchmarks/common/include/visc.h
@@ -96,6 +96,7 @@ void* __visc__tensor_add(void*, void*);
 void* __visc__tensor_mul(void*, void*);
 void* __visc__tensor_convolution(void*, void*, int, int, int, int);
 void* __visc__tensor_group_convolution(void*, void*, int, int, int, int, int, int);
+void* __visc__tensor_batchnorm(void*, void*, void*, void*, void*, double);  
 void* __visc__tensor_pool_max(void*, int, int, int, int, int, int);
 void* __visc__tensor_pool_mean(void*, int, int, int, int, int, int);
 void* __visc__tensor_relu(void*);