From 2215177e349c8587ba43d4c6ca075feb7e3c8705 Mon Sep 17 00:00:00 2001
From: liuqi <liuqi10@xiaomi.com>
Date: Thu, 2 Aug 2018 14:58:21 +0800
Subject: [PATCH] Add if_opencl_enabled_str for build static library without
 opencl.

---
 mace/libmace/BUILD                                   | 8 +++++---
 mace/mace.bzl                                        | 6 ++++++
 repository/opencl-kernel/opencl_kernel_configure.bzl | 3 +--
 3 files changed, 12 insertions(+), 5 deletions(-)

diff --git a/mace/libmace/BUILD b/mace/libmace/BUILD
index f96cce05..e8339220 100644
--- a/mace/libmace/BUILD
+++ b/mace/libmace/BUILD
@@ -15,6 +15,7 @@ load(
     "if_android_armv7",
     "if_hexagon_enabled",
     "if_opencl_enabled",
+    "if_opencl_enabled_str",
 )
 
 cc_library(
@@ -67,7 +68,6 @@ cc_library(
 genrule(
     name = "libmace_static",
     srcs = [
-        "//mace/codegen:generated_opencl",
         "//mace/codegen:generated_version",
         "//mace/core",
         "//mace/kernels",
@@ -76,11 +76,12 @@ genrule(
         "//mace/utils",
         "//mace/proto:mace_cc",
         "@com_google_protobuf//:protobuf_lite",
-    ],
+    ] + if_opencl_enabled([
+        "//mace/codegen:generated_opencl",
+    ]),
     outs = ["libmace.a"],
     cmd = "tmp_mri_file=$$(mktemp mace-static-lib-mri.XXXXXXXXXX);" +
           "mri_stream=$$(python $(location //mace/python/tools:archive_static_lib) " +
-          "$(locations //mace/codegen:generated_opencl) " +
           "$(locations //mace/codegen:generated_version) " +
           "$(locations //mace/core:core) " +
           "$(locations //mace/kernels:kernels) " +
@@ -89,6 +90,7 @@ genrule(
           "$(locations //mace/utils:utils) " +
           "$(locations //mace/proto:mace_cc) " +
           "$(locations @com_google_protobuf//:protobuf_lite) " +
+          if_opencl_enabled_str("$(locations //mace/codegen:generated_opencl) ") +
           "$@ " +
           "$$tmp_mri_file);" +
           "$(AR) -M <$$tmp_mri_file;" +
diff --git a/mace/mace.bzl b/mace/mace.bzl
index a88ac23e..3b91e44e 100644
--- a/mace/mace.bzl
+++ b/mace/mace.bzl
@@ -54,6 +54,12 @@ def if_opencl_enabled(a):
       "//conditions:default": [],
   })
 
+def if_opencl_enabled_str(a):
+  return select({
+      "//mace:opencl_enabled": a,
+      "//conditions:default": "",
+  })
+
 def mace_version_genrule():
   native.genrule(
       name = "mace_version_gen",
diff --git a/repository/opencl-kernel/opencl_kernel_configure.bzl b/repository/opencl-kernel/opencl_kernel_configure.bzl
index f5127ace..0da8838d 100644
--- a/repository/opencl-kernel/opencl_kernel_configure.bzl
+++ b/repository/opencl-kernel/opencl_kernel_configure.bzl
@@ -6,6 +6,7 @@ def _opencl_encrypt_kernel_impl(repository_ctx):
       Label("//repository/opencl-kernel:BUILD.tpl"))
 
   mace_root_path = str(repository_ctx.path(Label("@mace//:BUILD")))[:-len("BUILD")]
+  generated_files_path = repository_ctx.path("gen")
 
   ret = repository_ctx.execute(
       ["test", "-f", "%s/.git/logs/HEAD" % mace_root_path])
@@ -46,8 +47,6 @@ def _opencl_encrypt_kernel_impl(repository_ctx):
     unused_var = repository_ctx.path(Label("//:mace/kernels/opencl/cl/space_to_batch.cl"))
     unused_var = repository_ctx.path(Label("//:mace/kernels/opencl/cl/winograd_transform.cl"))
 
-  generated_files_path = repository_ctx.path("gen")
-
   python_bin_path = repository_ctx.which("python")
 
   repository_ctx.execute([
-- 
GitLab