Skip to content

Commit 832a726

Browse files
committed
fix(moore): retain generated ATen instantiations
1 parent f8e16ae commit 832a726

1 file changed

Lines changed: 5 additions & 1 deletion

File tree

src/CMakeLists.txt

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -720,11 +720,15 @@ if(WITH_TORCH)
720720
list(APPEND _torch_vendor_include_flags "-I${MUSA_ROOT}/include")
721721
endif()
722722

723-
# Vendor-specific defines required by forked `torch` headers.
723+
# Select vendor-specific generated implementations and any definitions
724+
# required by forked `torch` headers.
724725
set(_torch_extra_flags "")
725726
if(WITH_METAX)
726727
list(APPEND _torch_extra_flags "-DUSE_MACA=1" "-DWITH_METAX=1")
727728
endif()
729+
if(WITH_MOORE)
730+
list(APPEND _torch_extra_flags "-DWITH_MOORE=1")
731+
endif()
728732
if(WITH_CPU)
729733
list(APPEND _torch_extra_flags "-DWITH_CPU=1")
730734
endif()

0 commit comments

Comments
 (0)