iOS build cleanup.

PiperOrigin-RevId: 179256571
2017-12-15 16:49:30 -08:00 · 2017-12-15 16:49:30 -08:00 · 4bdea2e402
commit 4bdea2e402
parent dcb0666a2b
1 changed files with 7 additions and 0 deletions
--- a/tensorflow/tensorflow.bzl
+++ b/tensorflow/tensorflow.bzl
@ -103,6 +103,12 @@ def if_ios(a):
      "//conditions:default": [],
  })

+def if_ios_x86_64(a):
+  return select({
+      clean_dep("//tensorflow:ios_x86_64"): a,
+      "//conditions:default": [],
+  })
+
 def if_mobile(a):
  return select({
      clean_dep("//tensorflow:android"): a,
@ -177,6 +183,7 @@ def tf_copts(android_optimization_level_override="-O2"):
      + if_mkl(["-DINTEL_MKL=1", "-DEIGEN_USE_VML", "-fopenmp",])
      + if_android_arm(["-mfpu=neon"])
      + if_linux_x86_64(["-msse3"])
+      + if_ios_x86_64(["-msse4.1"])
      + select({
            clean_dep("//tensorflow:android"): android_copts,
            clean_dep("//tensorflow:darwin"): [],