1 files changed, 14 insertions, 3 deletions

diff --git a/tools/perf/util/c++/clang-test.cpp b/tools/perf/util/c++/clang-test.cpp
index 0f484fbb2b58..d84e760d2aab 100644
--- a/tools/perf/util/c++/clang-test.cpp
+++ b/tools/perf/util/c++/clang-test.cpp
@@ -3,6 +3,10 @@
 #include "llvm/IR/Function.h"
 #include "llvm/IR/LLVMContext.h"
 
+#include <util-cxx.h>
+#include <tests/llvm.h>
+#include <string>
+
 class perf_clang_scope {
 public:
         explicit perf_clang_scope() {perf_clang__init();}
@@ -14,17 +18,24 @@ extern "C" {
 int test__clang_to_IR(void)
 {
         perf_clang_scope _scope;
+        unsigned int kernel_version;
+
+        if (fetch_kernel_version(&kernel_version, NULL, 0))
+                return -1;
+
+        std::string cflag_kver("-DLINUX_VERSION_CODE=" +
+                                std::to_string(kernel_version));
 
         std::unique_ptr<llvm::Module> M =
-                perf::getModuleFromSource({"-DRESULT=1"},
+                perf::getModuleFromSource({cflag_kver.c_str()},
                                           "perf-test.c",
-                                          "int myfunc(void) {return RESULT;}");
+                                          test_llvm__bpf_base_prog);
 
         if (!M)
                 return -1;
 
         for (llvm::Function& F : *M)
-                if (F.getName() == "myfunc")
+                if (F.getName() == "bpf_func__SyS_epoll_wait")
                         return 0;
         return -1;
 }