Merge pull request #20538 from rongjiecomputer:png

PiperOrigin-RevId: 208509699

1 files changed, 17 insertions, 1 deletions

diff --git a/third_party/png.BUILD b/third_party/png.BUILD
index 17c5449cc0..c26a289717 100644
--- a/third_party/png.BUILD
+++ b/third_party/png.BUILD
@@ -29,6 +29,10 @@ cc_library(
         "pngwtran.c",
         "pngwutil.c",
     ] + select({
+        ":windows": [
+            "intel/intel_init.c",
+            "intel/filter_sse2_intrinsics.c",
+        ],
         "@org_tensorflow//tensorflow:linux_ppc64le": [
             "powerpc/powerpc_init.c",
             "powerpc/filter_vsx_intrinsics.c",
@@ -41,7 +45,14 @@ cc_library(
         "pngconf.h",
     ],
     includes = ["."],
-    linkopts = ["-lm"],
+    copts = select({
+        ":windows": ["-DPNG_INTEL_SSE_OPT=1"],
+        "//conditions:default": [],
+    }),
+    linkopts = select({
+        ":windows": [],
+        "//conditions:default": ["-lm"],
+    }),
     visibility = ["//visibility:public"],
     deps = ["@zlib_archive//:zlib"],
 )
@@ -52,3 +63,8 @@ genrule(
     outs = ["pnglibconf.h"],
     cmd = "sed -e 's/PNG_ZLIB_VERNUM 0/PNG_ZLIB_VERNUM 0x12b0/' $< >$@",
 )
+
+config_setting(
+    name = "windows",
+    values = {"cpu": "x64_windows"},
+)