mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 12:54:11 +08:00
b90aa18569
Re-land of https://github.com/pytorch/pytorch/pull/125242 Pull Request resolved: https://github.com/pytorch/pytorch/pull/127034 Approved by: https://github.com/malfet
9 lines
352 B
CMake
9 lines
352 B
CMake
# Build separate libraries the define custom classes/operators used from our Python tests.
|
|
# These are intended to be used with torch.ops.load_library() in our Python test suite.
|
|
add_library(aoti_custom_ops SHARED custom_ops.cpp)
|
|
target_link_libraries(aoti_custom_ops torch)
|
|
|
|
if(INSTALL_TEST)
|
|
install(TARGETS aoti_custom_ops DESTINATION lib)
|
|
endif()
|