diff --git a/BUILD.gn b/BUILD.gn
index 9b414f00a3ea3db3b26927b2538f202812bb701d..874ca22f232c532443859f2f2c9e993fcaea9e4b 100644
--- a/BUILD.gn
+++ b/BUILD.gn
@@ -13,6 +13,18 @@
 
 import("//build/ohos.gni")
 
+declare_args() {
+  neural_network_runtime_coverage = false
+}
+
+config("coverage_flags") {
+  if (neural_network_runtime_coverage) {
+    cflags = [ "--coverage" ]
+    cflags_cc = [ "--coverage" ]
+    ldflags = [ "--coverage" ]
+  }
+}
+
 group("nnrt_target") {
   deps = [
     "frameworks/native/neural_network_core:libneural_network_core",
diff --git a/bundle.json b/bundle.json
index d995206ec4d8201f5796e21a2952367f6aa980cb..b425acb10093867f385ee1b8aed9618ff5e0a073 100644
--- a/bundle.json
+++ b/bundle.json
@@ -64,6 +64,7 @@
                 }
             ],
             "test": [
+                "//foundation/ai/neural_network_runtime/test/unittest:unittest",
                 "//foundation/ai/neural_network_runtime:nnrt_fuzztest"
             ]
         }
diff --git a/frameworks/native/neural_network_core/BUILD.gn b/frameworks/native/neural_network_core/BUILD.gn
index 13a32345b3061f5128ee208181e2e23d103a5377..c8390e23202b988ed30430829f7cb9e78836c0e4 100644
--- a/frameworks/native/neural_network_core/BUILD.gn
+++ b/frameworks/native/neural_network_core/BUILD.gn
@@ -44,7 +44,10 @@ ohos_shared_library("libneural_network_core") {
 
   include_dirs = [ "../../.." ]
 
-  public_configs = [ ":nnrt_config" ]
+  public_configs = [
+    "../../..:coverage_flags",
+    ":nnrt_config",
+  ]
 
   external_deps = [
     "c_utils:utils",
diff --git a/frameworks/native/neural_network_runtime/BUILD.gn b/frameworks/native/neural_network_runtime/BUILD.gn
index 728286e3bd5395f2f3cc3bbdacd13ec2b6da3416..b59b9f1350fbd148c81b85eb73119f73c2d815bb 100644
--- a/frameworks/native/neural_network_runtime/BUILD.gn
+++ b/frameworks/native/neural_network_runtime/BUILD.gn
@@ -175,7 +175,10 @@ ohos_shared_library("libneural_network_runtime") {
 
   include_dirs = [ "../../.." ]
 
-  public_configs = [ ":nnrt_config" ]
+  public_configs = [
+    "../../..:coverage_flags",
+    ":nnrt_config",
+  ]
 
   external_deps = [
     "c_utils:utils",
diff --git a/test/unittest/BUILD.gn b/test/unittest/BUILD.gn
index eed3e38723fadf25cc0a701bafd17c7e40703bd5..1f2cee56e60ba4c3d52c38730036a685f5245e9a 100644
--- a/test/unittest/BUILD.gn
+++ b/test/unittest/BUILD.gn
@@ -15,9 +15,5 @@ import("//build/ohos.gni")
 
 group("unittest") {
   testonly = true
-  deps = [
-    "components:components_unittest",
-    "inner_kits:inner_kits_unittest",
-    "ops:ops_unittest",
-  ]
+  deps = [ "ops:ops_unittest" ]
 }
diff --git a/test/unittest/ops/BUILD.gn b/test/unittest/ops/BUILD.gn
index a3cc2c35945e853d0dc56fa1fd5e8927cca64afe..8a9dab0770d53b3a64a7c155635a019a5f8ade5a 100644
--- a/test/unittest/ops/BUILD.gn
+++ b/test/unittest/ops/BUILD.gn
@@ -149,7 +149,10 @@ ohos_unittest("OpsUnittest") {
   sources += [ "./ops_test.cpp" ]
   sources += [ "../common/base_test.cpp" ]
 
-  configs = [ ":module_private_config" ]
+  configs = [
+    "../../..:coverage_flags",
+    ":module_private_config",
+  ]
 
   deps = [
     "../../../frameworks/native/neural_network_core:libneural_network_core",