fix(//cpp/trtorchc): Support building trtorchc with the pre_cxx11_abi

for libtrtorch building with --config=pre_cxx11_abi Signed-off-by: Naren Dasan <[email protected]> Signed-off-by: Naren Dasan <[email protected]>
pytorch · Jul 8, 2020 · 172d4d5 · 172d4d5
1 parent 58006ba
commit 172d4d5
Showing 1 changed file with 18 additions and 4 deletions.
diff --git a/cpp/trtorchc/BUILD b/cpp/trtorchc/BUILD
@@ -1,14 +1,28 @@
 package(default_visibility = ["//visibility:public"])
 
+config_setting(
+    name = "use_pre_cxx11_abi",
+    values = {
+        "define": "abi=pre_cxx11_abi",
+    }
+)
+
 cc_binary(
     name = "trtorchc",
     srcs = [
         "main.cpp"
     ],
     deps = [
-        "@libtorch//:libtorch",
-        "@libtorch//:caffe2",
         "//third_party/args",
         "//cpp/api:trtorch"
-    ],
-)
+    ] + select({
+        ":use_pre_cxx11_abi":  [
+            "@libtorch_pre_cxx11_abi//:libtorch",
+            "@libtorch_pre_cxx11_abi//:caffe2",
+        ],
+        "//conditions:default":  [
+            "@libtorch//:libtorch",
+            "@libtorch//:caffe2",
+        ],
+    }),
+)