1if(USE_DISTRIBUTED AND NOT WIN32) 2 set(DIST_AUTOGRAD_TEST_DIR "${TORCH_ROOT}/test/cpp/dist_autograd") 3 set(DIST_AUTOGRAD_TEST_SOURCES 4 ${TORCH_ROOT}/test/cpp/common/main.cpp 5 ${DIST_AUTOGRAD_TEST_DIR}/test_dist_autograd.cpp 6 ) 7 8 add_executable(test_dist_autograd ${DIST_AUTOGRAD_TEST_SOURCES}) 9 target_include_directories(test_dist_autograd PRIVATE ${ATen_CPU_INCLUDE}) 10 target_link_libraries(test_dist_autograd PRIVATE torch gtest) 11 12 if(USE_CUDA) 13 target_compile_definitions(test_dist_autograd PRIVATE "USE_CUDA") 14 endif() 15 16 if(INSTALL_TEST) 17 install(TARGETS test_dist_autograd DESTINATION bin) 18 # Install PDB files for MSVC builds 19 if(MSVC AND BUILD_SHARED_LIBS) 20 install(FILES $<TARGET_PDB_FILE:test_dist_autograd> DESTINATION bin OPTIONAL) 21 endif() 22 endif() 23endif() 24