Using protoc

JuliaPackaging · Feb 21, 2022 · c6b415e · c6b415e
1 parent 2b0d17e
commit c6b415e
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/T/Torch/Torch/build_tarballs.jl b/T/Torch/Torch/build_tarballs.jl
@@ -23,7 +23,10 @@ cmake \
     -DCMAKE_BUILD_TYPE=Release \
     -DCMAKE_INSTALL_PREFIX=$prefix \
     -DCMAKE_TOOLCHAIN_FILE=${CMAKE_TARGET_TOOLCHAIN} \
+    -DBUILD_CUSTOM_PROTOBUF=OFF \
     -DBUILD_SHARED_LIBS=ON \
+    -DPROTOBUF_PROTOC_EXECUTABLE=$host_bindir/protoc \
+    -DCAFFE2_CUSTOM_PROTOC_EXECUTABLE=$host_bindir/protoc \
     -DPYTHON_EXECUTABLE=`which python3` \
     ..
 cmake --build .
@@ -57,7 +60,8 @@ products = [
 
 # Dependencies that must be installed before this package can be built
 dependencies = [
-    Dependency("MKL_jll"; platforms = mkl_platforms)
+    Dependency("MKL_jll"; platforms = mkl_platforms),
+    HostBuildDependency("protoc_jll")
 ]
 
 # Build the tarballs, and possibly a `build.jl` as well.