wehub-resource-sync
233 lines
5.4 KiB
CMake
233 lines
5.4 KiB
CMake
cc_test(
|
|
memory_stats_test
|
|
SRCS memory_stats_test.cc
|
|
DEPS)
|
|
cc_test(
|
|
stats_test
|
|
SRCS stats_test.cc
|
|
DEPS)
|
|
|
|
cc_test(
|
|
naive_best_fit_allocator_test
|
|
SRCS naive_best_fit_allocator_test.cc
|
|
DEPS phi common)
|
|
cc_test(
|
|
buffered_allocator_test
|
|
SRCS buffered_allocator_test.cc
|
|
DEPS phi common)
|
|
|
|
if(WITH_GPU)
|
|
nv_test(
|
|
thread_local_allocator_test
|
|
SRCS thread_local_allocator_test.cc
|
|
DEPS phi common)
|
|
endif()
|
|
if(WITH_ROCM)
|
|
hip_test(
|
|
thread_local_allocator_test
|
|
SRCS thread_local_allocator_test.cc
|
|
DEPS phi common)
|
|
endif()
|
|
|
|
if(WITH_GPU)
|
|
nv_test(
|
|
best_fit_allocator_test
|
|
SRCS best_fit_allocator_test.cc best_fit_allocator_test.cu
|
|
DEPS phi common)
|
|
elseif(WITH_ROCM)
|
|
hip_test(
|
|
best_fit_allocator_test
|
|
SRCS best_fit_allocator_test.cc best_fit_allocator_test.cu
|
|
DEPS phi common)
|
|
else()
|
|
if(WIN32)
|
|
cc_test(
|
|
best_fit_allocator_test
|
|
SRCS best_fit_allocator_test.cc
|
|
DEPS type_info common)
|
|
else()
|
|
cc_test(
|
|
best_fit_allocator_test
|
|
SRCS best_fit_allocator_test.cc
|
|
DEPS phi common)
|
|
endif()
|
|
endif()
|
|
|
|
cc_test(
|
|
test_aligned_allocator
|
|
SRCS test_aligned_allocator.cc
|
|
DEPS phi common)
|
|
|
|
if(WITH_GPU)
|
|
if(WIN32)
|
|
message(STATUS "Skip allocator_visitor_test on Windows")
|
|
else()
|
|
nv_test(
|
|
allocator_visitor_test
|
|
SRCS allocator_visitor_test.cc
|
|
DEPS phi common)
|
|
endif()
|
|
endif()
|
|
|
|
if(WIN32)
|
|
cc_test(
|
|
retry_allocator_test
|
|
SRCS retry_allocator_test.cc
|
|
DEPS type_info common)
|
|
else()
|
|
cc_test(
|
|
retry_allocator_test
|
|
SRCS retry_allocator_test.cc
|
|
DEPS phi common)
|
|
endif()
|
|
if(TEST retry_allocator_test)
|
|
set_tests_properties(retry_allocator_test PROPERTIES LABELS
|
|
"RUN_TYPE=EXCLUSIVE")
|
|
endif()
|
|
|
|
cc_test(
|
|
allocator_facade_abs_flags_test
|
|
SRCS allocator_facade_abs_flags_test.cc
|
|
DEPS phi common)
|
|
|
|
if(WITH_GPU)
|
|
if(WIN32)
|
|
message(STATUS "Skip vmm_allocator_tail_test on Windows")
|
|
else()
|
|
nv_test(
|
|
vmm_allocator_tail_test
|
|
SRCS vmm_allocator_tail_test.cc
|
|
DEPS phi common)
|
|
nv_test(
|
|
cuda_virtual_mem_allocator_v2_test
|
|
SRCS cuda_virtual_mem_allocator_v2_test.cu
|
|
DEPS phi common)
|
|
endif()
|
|
endif()
|
|
|
|
cc_test(
|
|
allocator_facade_frac_flags_test
|
|
SRCS allocator_facade_frac_flags_test.cc
|
|
DEPS phi common)
|
|
|
|
if(WITH_GPU)
|
|
nv_test(
|
|
malloc_test
|
|
SRCS malloc_test.cu
|
|
DEPS phi)
|
|
nv_test(
|
|
stream_safe_cuda_alloc_test
|
|
SRCS stream_safe_cuda_alloc_test.cu
|
|
DEPS phi common)
|
|
nv_test(
|
|
cuda_managed_memory_test
|
|
SRCS cuda_managed_memory_test.cu
|
|
DEPS phi common)
|
|
nv_test(
|
|
cuda_malloc_async_allocator_test
|
|
SRCS cuda_malloc_async_allocator_test.cu
|
|
DEPS phi common)
|
|
if(WITH_TESTING AND TEST stream_safe_cuda_alloc_test)
|
|
set_tests_properties(
|
|
stream_safe_cuda_alloc_test
|
|
PROPERTIES ENVIRONMENT "FLAGS_use_stream_safe_cuda_allocator=true; \
|
|
FLAGS_allocator_strategy=auto_growth")
|
|
endif()
|
|
if(WITH_TESTING AND TEST cuda_malloc_async_allocator_test)
|
|
set_tests_properties(
|
|
cuda_malloc_async_allocator_test
|
|
PROPERTIES ENVIRONMENT "FLAGS_use_cuda_malloc_async_allocator=true")
|
|
endif()
|
|
endif()
|
|
|
|
if(WITH_ROCM)
|
|
hip_test(
|
|
malloc_test
|
|
SRCS malloc_test.cu
|
|
DEPS phi)
|
|
hip_test(
|
|
cuda_managed_memory_test
|
|
SRCS cuda_managed_memory_test.cu
|
|
DEPS phi common)
|
|
endif()
|
|
|
|
if(WITH_TESTING AND TEST cuda_managed_memory_test)
|
|
set_tests_properties(
|
|
cuda_managed_memory_test
|
|
PROPERTIES
|
|
ENVIRONMENT
|
|
"FLAGS_use_cuda_managed_memory=true;FLAGS_use_cuda_malloc_async_allocator=false;FLAGS_allocator_strategy=auto_growth"
|
|
TIMEOUT
|
|
50)
|
|
endif()
|
|
|
|
if(WITH_GPU AND WITH_TESTING)
|
|
nv_test(
|
|
get_base_ptr_test
|
|
SRCS get_base_ptr_test.cu
|
|
DEPS phi common)
|
|
set_tests_properties(
|
|
get_base_ptr_test
|
|
PROPERTIES ENVIRONMENT "FLAGS_allocator_strategy=auto_growth;
|
|
FLAGS_use_stream_safe_cuda_allocator=true;")
|
|
endif()
|
|
|
|
if(WIN32)
|
|
cc_test(
|
|
auto_growth_best_fit_allocator_facade_test
|
|
SRCS auto_growth_best_fit_allocator_facade_test.cc
|
|
DEPS type_info common)
|
|
else()
|
|
cc_test(
|
|
auto_growth_best_fit_allocator_facade_test
|
|
SRCS auto_growth_best_fit_allocator_facade_test.cc
|
|
DEPS phi common)
|
|
endif()
|
|
cc_test(
|
|
auto_growth_best_fit_allocator_test
|
|
SRCS auto_growth_best_fit_allocator_test.cc
|
|
DEPS phi common)
|
|
|
|
if(NOT WIN32)
|
|
cc_test(
|
|
mmap_allocator_test
|
|
SRCS mmap_allocator_test.cc
|
|
DEPS phi common)
|
|
endif()
|
|
|
|
cc_test(
|
|
system_allocator_test
|
|
SRCS system_allocator_test.cc
|
|
DEPS phi common)
|
|
|
|
if(WITH_GPU)
|
|
if(WIN32)
|
|
message(STATUS "Skip vmm_auto_growth_best_fit_allocator_test on Windows")
|
|
else()
|
|
nv_test(
|
|
vmm_auto_growth_best_fit_allocator_test
|
|
SRCS vmm_auto_growth_best_fit_allocator_test.cu
|
|
DEPS phi common)
|
|
nv_test(
|
|
vmm_auto_growth_best_fit_allocator_v2_test
|
|
SRCS vmm_auto_growth_best_fit_allocator_v2_test.cu
|
|
DEPS phi common)
|
|
if(WITH_TESTING AND TEST vmm_auto_growth_best_fit_allocator_test)
|
|
set_tests_properties(
|
|
vmm_auto_growth_best_fit_allocator_test
|
|
PROPERTIES ENVIRONMENT "FLAGS_use_virtual_memory_auto_growth=true")
|
|
endif()
|
|
endif()
|
|
endif()
|
|
|
|
if(WITH_GPU)
|
|
if(WIN32)
|
|
message(STATUS "Skip multi_scale_allocator_test on Windows")
|
|
else()
|
|
nv_test(
|
|
multi_scale_allocator_test
|
|
SRCS multi_scale_allocator_test.cc
|
|
DEPS phi common)
|
|
endif()
|
|
endif()
|