mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
3f5ed05688
- Solves a problem related to .hip source files being ignored by the build system when HIP language is not enabled in CMake. - Also ensures that the test executables link to an appropriate CRT Runtime Library and hence have access to all the necessary symbols. Previously, there were many problems related to linkage errors. - Moves part of Linux-related hipBLASLt changes in `LoadHIP.cmake` under the UNIX conditional branch, as these aren't supported on Windows yet. Pull Request resolved: https://github.com/pytorch/pytorch/pull/146599 Approved by: https://github.com/jeffdaily
31 lines
1.4 KiB
CMake
31 lines
1.4 KiB
CMake
# ---[ Test binaries.
|
|
|
|
set(C10_CUDA_ALL_TEST_FILES
|
|
impl/CUDAAssertionsTest_1_var_test.cu
|
|
impl/CUDAAssertionsTest_catches_stream.cu
|
|
impl/CUDAAssertionsTest_catches_thread_and_block_and_device.cu
|
|
impl/CUDAAssertionsTest_from_2_processes.cu
|
|
impl/CUDAAssertionsTest_multiple_writes_from_blocks_and_threads.cu
|
|
impl/CUDAAssertionsTest_multiple_writes_from_multiple_blocks.cu
|
|
impl/CUDAAssertionsTest_multiple_writes_from_same_block.cu
|
|
impl/CUDATest.cpp
|
|
)
|
|
if(BUILD_TEST)
|
|
foreach(test_src ${C10_CUDA_ALL_TEST_FILES})
|
|
get_filename_component(test_file_name ${test_src} NAME_WE)
|
|
set(test_name "c10_cuda_${test_file_name}")
|
|
if(WIN32 AND test_src MATCHES "^.*\.hip$")
|
|
set_source_files_properties(${test_src} PROPERTIES HIP_SOURCE_PROPERTY_FORMAT 1)
|
|
hip_add_executable(${test_name} "${test_src}")
|
|
set_target_properties(${test_name} PROPERTIES LINKER_LANGUAGE CXX HIP_ARCHITECTURES ${PYTORCH_ROCM_ARCH})
|
|
else()
|
|
add_executable(${test_name} "${test_src}")
|
|
endif()
|
|
target_link_libraries(${test_name} ${C10_CUDA_LIB} ${C10_LIB} gtest_main)
|
|
add_test(NAME ${test_name} COMMAND $<TARGET_FILE:${test_name}>)
|
|
if(INSTALL_TEST)
|
|
set_target_properties(${test_name} PROPERTIES INSTALL_RPATH "${CMAKE_INSTALL_RPATH}:${_rpath_portable_origin}/../lib")
|
|
install(TARGETS ${test_name} DESTINATION test)
|
|
endif()
|
|
endforeach()
|
|
endif()
|