chore: import upstream snapshot with attribution
This commit is contained in:
wehub-resource-sync
@@ -0,0 +1,232 @@
|
||||
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()
|
||||
Reference in New Issue
Block a user