mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 12:54:11 +08:00
493c7fa66f
By turning on compatibility mode for protobuf, nnpack, PSimd and FP16, ittapi, TensorPipe and Gloo Update CMake requirements Revert 0ece461ccafe5649d2d0f058ff5477765fd56499 and b0901d62ae2c2e909f91401eacebf3731df20cbe to test that it actually works TODO: - Update/get rid of those libraries Fixes https://github.com/pytorch/pytorch/issues/150149 Pull Request resolved: https://github.com/pytorch/pytorch/pull/150203 Approved by: https://github.com/clee2000
18 lines
536 B
CMake
18 lines
536 B
CMake
# Basic CMake setup
|
|
cmake_minimum_required(VERSION 3.15 FATAL_ERROR)
|
|
project(custom_backend)
|
|
|
|
if(USE_ROCM)
|
|
include(utils)
|
|
include(LoadHIP)
|
|
endif()
|
|
find_package(Torch REQUIRED)
|
|
|
|
add_library(custom_backend SHARED custom_backend.cpp)
|
|
set_property(TARGET custom_backend PROPERTY CXX_STANDARD 17)
|
|
target_link_libraries(custom_backend "${TORCH_LIBRARIES}")
|
|
|
|
add_executable(test_custom_backend test_custom_backend.cpp)
|
|
set_property(TARGET test_custom_backend PROPERTY CXX_STANDARD 17)
|
|
target_link_libraries(test_custom_backend custom_backend)
|