diff --git a/llvm/test/VISC/DNN_Benchmarks/common/include/visc.h b/llvm/test/VISC/DNN_Benchmarks/common/include/visc.h
index f7d534a5d9b41480b893ef2c6fe839bba3dfe551..2b778e8848255f84a84f4e3d560a6cb82fbe439c 100644
--- a/llvm/test/VISC/DNN_Benchmarks/common/include/visc.h
+++ b/llvm/test/VISC/DNN_Benchmarks/common/include/visc.h
@@ -15,10 +15,8 @@
 #ifdef __cplusplus
 extern "C" {
 void __visc__hint(visc::Target);
-void __visc__wait(void*);
 #else
 void __visc__hint(enum Target);
-void __visc__wait(unsigned);
 #endif
 
 #ifdef __cplusplus
@@ -40,6 +38,7 @@ void* __visc__edge(void*, void*, unsigned, unsigned, unsigned, unsigned);
 void __visc__push(void*, void*);
 void* __visc__pop(void*);
 void* __visc__launch(unsigned, ...);
+void __visc__wait(void*);
 
 void* __visc__getNode();
 void* __visc__getParentNode(void*);
diff --git a/llvm/test/VISC/parboil/common/include/visc.h b/llvm/test/VISC/parboil/common/include/visc.h
index f25c3cc4dcf090b6a057c369da732b80895788d0..684cda52267a5f03a0bf332afb99d183a2357a3a 100644
--- a/llvm/test/VISC/parboil/common/include/visc.h
+++ b/llvm/test/VISC/parboil/common/include/visc.h
@@ -15,10 +15,8 @@
 #ifdef __cplusplus
 extern "C" {
 void __visc__hint(visc::Target);
-void __visc__wait(void*);
 #else
 void __visc__hint(enum Target);
-void __visc__wait(unsigned);
 #endif
 
 #ifdef __cplusplus
@@ -40,6 +38,7 @@ void* __visc__edge(void*, void*, unsigned, unsigned, unsigned, unsigned);
 void __visc__push(void*, void*);
 void* __visc__pop(void*);
 void* __visc__launch(unsigned, ...);
+void __visc__wait(void*);
 
 void* __visc__getNode();
 void* __visc__getParentNode(void*);