We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent f8e16ae commit 832a726Copy full SHA for 832a726
1 file changed
src/CMakeLists.txt
@@ -720,11 +720,15 @@ if(WITH_TORCH)
720
list(APPEND _torch_vendor_include_flags "-I${MUSA_ROOT}/include")
721
endif()
722
723
- # Vendor-specific defines required by forked `torch` headers.
+ # Select vendor-specific generated implementations and any definitions
724
+ # required by forked `torch` headers.
725
set(_torch_extra_flags "")
726
if(WITH_METAX)
727
list(APPEND _torch_extra_flags "-DUSE_MACA=1" "-DWITH_METAX=1")
728
729
+ if(WITH_MOORE)
730
+ list(APPEND _torch_extra_flags "-DWITH_MOORE=1")
731
+ endif()
732
if(WITH_CPU)
733
list(APPEND _torch_extra_flags "-DWITH_CPU=1")
734
0 commit comments