diff --git a/BUILD.gn b/BUILD.gn
index d9e96d39f02a86239d3bd1694fe3d6980db0eaf5..b62be659ae5d6746cb7ada5f8b8d4ae165dfeebe 100644
--- a/BUILD.gn
+++ b/BUILD.gn
@@ -23,8 +23,10 @@ config("cl_config") {
 
 config("cl_public_config") {
   include_dirs = [
-    "./",
+    "CL",
     "include",
+    "src",
+    "tests",
   ]
 }
 
@@ -41,7 +43,12 @@ ohos_shared_library("libcl") {
 }
 
 config("opencl_headers_public_config") {
-  include_dirs = [ "./" ]
+    include_dirs = [
+    "CL",
+    "include",
+    "src",
+    "tests",
+  ]
 }
 
 source_set("opencl_headers") {