Skip to content

Commit

Permalink
Update conanfile.py
Browse files Browse the repository at this point in the history
  • Loading branch information
joda01 authored Jan 26, 2025
1 parent 267830c commit 06a5b94
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion recipes/libtorch/all/conanfile.py
Original file line number Diff line number Diff line change
Expand Up @@ -632,6 +632,9 @@ def _itt():
def _fp16():
return ["fp16::fp16"] if self.options.get_safe("with_qnnpack",False) == False else []

def _pybind()
return ["pybind11::pybind11"] if self.is_mac_os == False else []

self.cpp_info.set_property("cmake_file_name", "Torch")

# Export official CMake variables
Expand Down Expand Up @@ -669,7 +672,7 @@ def _fp16():
self.cpp_info.components["torch_cpu_link_order_workaround"].requires.extend(
["_headers", "c10", "eigen::eigen", "fmt::fmt", "foxi::foxi"] +
_fbgemm() + _sleef() + _onednn() + _protobuf() + _fbgemm() + _kineto() + _openblas() + _lapack() +
_vulkan() + _opencl() + _openmp() + _nnpack() + _xnnpack() + _qnnpack() + _itt() + _fp16()
_vulkan() + _opencl() + _openmp() + _nnpack() + _xnnpack() + _qnnpack() + _itt() + _fp16() + _pybind()
)
if self.settings.os == "Linux":
self.cpp_info.components["torch_cpu_link_order_workaround"].system_libs.extend(["dl", "m", "pthread", "rt"])
Expand Down

0 comments on commit 06a5b94

Please sign in to comment.