Update build.jinja

sail-sg · Dec 4, 2023 · 19ec135 · 19ec135
1 parent 9125526
commit 19ec135
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/maple2jax/libxc/build.jinja b/maple2jax/libxc/build.jinja
@@ -105,12 +105,22 @@ pybind_extension(
     visibility = ["//visibility:public"],
 )
 
+genrule(
+    name = "gen_libxc_pyd",
+    srcs = [":libxc.so"],
+    outs = ["libxc.pyd"],
+    cmd = "cp $< $@",
+)
+
 py_library(
     name = "libxc",
     srcs = glob(
         ["*.py"],
         exclude=["gen_build.py", "wrap.py"],
     ),
-    data = [":libxc.so"],
+    data = select({
+        "@platforms//os:windows": [":gen_libxc_pyd"],
+        "//conditions:default": [":libxc.so"],
+    }),
     visibility = ["//visibility:public"],
 )