Skip to content
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
165 changes: 81 additions & 84 deletions test/cpp/compat/CMakeLists.txt
Original file line number Diff line number Diff line change
@@ -1,92 +1,89 @@
if(NOT WIN32)
# c10 core tests (CPU compatible)
cc_test(c10_Device_test SRCS c10_Device_test.cc)
cc_test(c10_DispatchKeySet_test SRCS c10_DispatchKeySet_test.cc)
cc_test(c10_DispatchKey_test SRCS c10_DispatchKey_test.cc)
cc_test(c10_MemoryFormat_test SRCS c10_MemoryFormat_test.cc)
cc_test(c10_ScalarType_test SRCS c10_ScalarType_test.cc)
cc_test(c10_SizesAndStrides_test SRCS c10_SizesAndStrides_test.cc)
cc_test(c10_TensorOptions_test SRCS c10_TensorOptions_test.cc)
cc_test(c10_TypeMeta_test SRCS c10_TypeMeta_test.cc)
cc_test(c10_intrusive_ptr_lifecycle_test
SRCS c10_intrusive_ptr_lifecycle_test.cc)
cc_test(c10_layout_test SRCS c10_layout_test.cc)
cc_test(c10_ptr_test SRCS c10_ptr_test.cc)
cc_test(c10_storage_test SRCS c10_storage_test.cc)
# c10 core tests (CPU compatible)
cc_test(c10_Device_test SRCS c10_Device_test.cc)
cc_test(c10_DispatchKeySet_test SRCS c10_DispatchKeySet_test.cc)
cc_test(c10_DispatchKey_test SRCS c10_DispatchKey_test.cc)
cc_test(c10_MemoryFormat_test SRCS c10_MemoryFormat_test.cc)
cc_test(c10_ScalarType_test SRCS c10_ScalarType_test.cc)
cc_test(c10_SizesAndStrides_test SRCS c10_SizesAndStrides_test.cc)
cc_test(c10_TensorOptions_test SRCS c10_TensorOptions_test.cc)
cc_test(c10_TypeMeta_test SRCS c10_TypeMeta_test.cc)
cc_test(c10_intrusive_ptr_lifecycle_test
SRCS c10_intrusive_ptr_lifecycle_test.cc)
cc_test(c10_layout_test SRCS c10_layout_test.cc)
cc_test(c10_ptr_test SRCS c10_ptr_test.cc)
cc_test(c10_storage_test SRCS c10_storage_test.cc)

# ATen core tests (CPU compatible)
cc_test(ATen_all_test SRCS ATen_all_test.cc)
cc_test(ATen_any_test SRCS ATen_any_test.cc)
cc_test(ATen_as_strided_test SRCS ATen_as_strided_test.cc)
cc_test(ATen_autograd_test SRCS ATen_autograd_test.cc)
cc_test(ATen_chunk_test SRCS ATen_chunk_test.cc)
cc_test(ATen_clamp_test SRCS ATen_clamp_test.cc)
cc_test(ATen_coalesce_test SRCS ATen_coalesce_test.cc)
cc_test(ATen_dense_sparse_conversion_test
SRCS ATen_dense_sparse_conversion_test.cc)
cc_test(ATen_empty_test SRCS ATen_empty_test.cc)
cc_test(ATen_equal_test SRCS ATen_equal_test.cc)
cc_test(ATen_expand_test SRCS ATen_expand_test.cc)
cc_test(ATen_eye_test SRCS ATen_eye_test.cc)
cc_test(ATen_factory_default_dtype_test
SRCS ATen_factory_default_dtype_test.cc)
cc_test(ATen_flatten_test SRCS ATen_flatten_test.cc)
cc_test(ATen_from_blob_test SRCS ATen_from_blob_test.cc)
cc_test(ATen_hook_test SRCS ATen_hook_test.cc)
cc_test(ATen_index_test SRCS ATen_index_test.cc)
cc_test(ATen_item_test SRCS ATen_item_test.cc)
cc_test(ATen_narrow_test SRCS ATen_narrow_test.cc)
cc_test(ATen_new_test SRCS ATen_new_test.cc)
cc_test(ATen_nnz_test SRCS ATen_nnz_test.cc)
cc_test(ATen_rename_test SRCS ATen_rename_test.cc)
cc_test(ATen_reshape_test SRCS ATen_reshape_test.cc)
cc_test(ATen_resize_test SRCS ATen_resize_test.cc)
cc_test(ATen_squeeze_test SRCS ATen_squeeze_test.cc)
cc_test(ATen_std_var_test SRCS ATen_std_var_test.cc)
cc_test(ATen_sum_test SRCS ATen_sum_test.cc)
cc_test(ATen_t_test SRCS ATen_t_test.cc)
cc_test(ATen_tensor_data_test SRCS ATen_tensor_data_test.cc)
cc_test(ATen_toString_test SRCS ATen_toString_test.cc)
cc_test(ATen_to_test SRCS ATen_to_test.cc)
cc_test(ATen_transpose_test SRCS ATen_transpose_test.cc)
cc_test(ATen_Utils_test SRCS ATen_Utils_test.cc)
cc_test(ATen_values_test SRCS ATen_values_test.cc)
cc_test(ATen_viewAs_test SRCS ATen_viewAs_test.cc)
# ATen core tests (CPU compatible)
cc_test(ATen_all_test SRCS ATen_all_test.cc)
cc_test(ATen_any_test SRCS ATen_any_test.cc)
cc_test(ATen_as_strided_test SRCS ATen_as_strided_test.cc)
cc_test(ATen_autograd_test SRCS ATen_autograd_test.cc)
cc_test(ATen_chunk_test SRCS ATen_chunk_test.cc)
cc_test(ATen_clamp_test SRCS ATen_clamp_test.cc)
cc_test(ATen_coalesce_test SRCS ATen_coalesce_test.cc)
cc_test(ATen_dense_sparse_conversion_test
SRCS ATen_dense_sparse_conversion_test.cc)
cc_test(ATen_empty_test SRCS ATen_empty_test.cc)
cc_test(ATen_equal_test SRCS ATen_equal_test.cc)
cc_test(ATen_expand_test SRCS ATen_expand_test.cc)
cc_test(ATen_eye_test SRCS ATen_eye_test.cc)
cc_test(ATen_factory_default_dtype_test SRCS ATen_factory_default_dtype_test.cc)
cc_test(ATen_flatten_test SRCS ATen_flatten_test.cc)
cc_test(ATen_from_blob_test SRCS ATen_from_blob_test.cc)
cc_test(ATen_hook_test SRCS ATen_hook_test.cc)
cc_test(ATen_index_test SRCS ATen_index_test.cc)
cc_test(ATen_item_test SRCS ATen_item_test.cc)
cc_test(ATen_narrow_test SRCS ATen_narrow_test.cc)
cc_test(ATen_new_test SRCS ATen_new_test.cc)
cc_test(ATen_nnz_test SRCS ATen_nnz_test.cc)
cc_test(ATen_rename_test SRCS ATen_rename_test.cc)
cc_test(ATen_reshape_test SRCS ATen_reshape_test.cc)
cc_test(ATen_resize_test SRCS ATen_resize_test.cc)
cc_test(ATen_squeeze_test SRCS ATen_squeeze_test.cc)
cc_test(ATen_std_var_test SRCS ATen_std_var_test.cc)
cc_test(ATen_sum_test SRCS ATen_sum_test.cc)
cc_test(ATen_t_test SRCS ATen_t_test.cc)
cc_test(ATen_tensor_data_test SRCS ATen_tensor_data_test.cc)
cc_test(ATen_toString_test SRCS ATen_toString_test.cc)
cc_test(ATen_to_test SRCS ATen_to_test.cc)
cc_test(ATen_transpose_test SRCS ATen_transpose_test.cc)
cc_test(ATen_Utils_test SRCS ATen_Utils_test.cc)
cc_test(ATen_values_test SRCS ATen_values_test.cc)
cc_test(ATen_viewAs_test SRCS ATen_viewAs_test.cc)

# torch library tests (CPU compatible)
cc_test(torch_library_test SRCS torch_library_test.cc)
cc_test(torch_library_dispatch_test SRCS torch_library_dispatch_test.cc)
# torch library tests (CPU compatible)
cc_test(torch_library_test SRCS torch_library_test.cc)
cc_test(torch_library_dispatch_test SRCS torch_library_dispatch_test.cc)

# GPU-runtime compat tests are not fully audited on ROCm/DCU yet.
# Keep the DCU surface limited to the cases adapted in this PR.
if(WITH_ROCM)
cc_test(ATen_CUDAContext_test SRCS ATen_CUDAContext_test.cc)
cc_test(ATen_record_stream_test SRCS ATen_record_stream_test.cc)
cc_test(c10_Event_test SRCS c10_Event_test.cc)
cc_test(c10_Stream_test SRCS c10_Stream_test.cc)
else()
cc_test(ATen_TensorAccessor_test SRCS ATen_TensorAccessor_test.cc)
cc_test(ATen_basic_test SRCS ATen_basic_test.cc)
cc_test(ATen_local_scalar_dense_test SRCS ATen_local_scalar_dense_test.cc)
cc_test(ATen_memory_test SRCS ATen_memory_test.cc)
cc_test(ATen_pin_memory_creation_test SRCS ATen_pin_memory_creation_test.cc)
cc_test(ATen_record_stream_test SRCS ATen_record_stream_test.cc)
cc_test(ATen_select_test SRCS ATen_select_test.cc)
cc_test(ATen_split_test SRCS ATen_split_test.cc)
# GPU-runtime compat tests are not fully audited on ROCm/DCU yet.
# Keep the DCU surface limited to the cases adapted in this PR.
if(WITH_ROCM)
cc_test(ATen_CUDAContext_test SRCS ATen_CUDAContext_test.cc)
cc_test(ATen_record_stream_test SRCS ATen_record_stream_test.cc)
cc_test(c10_Event_test SRCS c10_Event_test.cc)
cc_test(c10_Stream_test SRCS c10_Stream_test.cc)
else()
cc_test(ATen_TensorAccessor_test SRCS ATen_TensorAccessor_test.cc)
cc_test(ATen_basic_test SRCS ATen_basic_test.cc)
cc_test(ATen_local_scalar_dense_test SRCS ATen_local_scalar_dense_test.cc)
cc_test(ATen_memory_test SRCS ATen_memory_test.cc)
cc_test(ATen_pin_memory_creation_test SRCS ATen_pin_memory_creation_test.cc)
cc_test(ATen_record_stream_test SRCS ATen_record_stream_test.cc)
cc_test(ATen_select_test SRCS ATen_select_test.cc)
cc_test(ATen_split_test SRCS ATen_split_test.cc)

cc_test(ATen_CUDAContext_test SRCS ATen_CUDAContext_test.cc)
cc_test(ATen_philox_test SRCS ATen_philox_test.cc)
cc_test(c10_Event_test SRCS c10_Event_test.cc)
cc_test(c10_Stream_test SRCS c10_Stream_test.cc)
endif()
cc_test(ATen_CUDAContext_test SRCS ATen_CUDAContext_test.cc)
cc_test(ATen_philox_test SRCS ATen_philox_test.cc)
cc_test(c10_Event_test SRCS c10_Event_test.cc)
cc_test(c10_Stream_test SRCS c10_Stream_test.cc)
endif()

if(WITH_GPU)
nv_test(ATen_CUDABlas_test SRCS ATen_CUDABlas_test.cc)
nv_test(ATen_cuda_test SRCS ATen_cuda_test.cc)
nv_test(c10_cuda_generator_test SRCS c10_cuda_generator_test.cc)
nv_test(c10_generator_impl_test SRCS c10_generator_impl_test.cc)
endif()
cc_test(schema_parser_type_test SRCS schema_parser_type_test.cc)
if(WITH_GPU)
nv_test(ATen_CUDABlas_test SRCS ATen_CUDABlas_test.cc)
nv_test(ATen_cuda_test SRCS ATen_cuda_test.cc)
nv_test(c10_cuda_generator_test SRCS c10_cuda_generator_test.cc)
nv_test(c10_generator_impl_test SRCS c10_generator_impl_test.cc)
endif()
cc_test(schema_parser_type_test SRCS schema_parser_type_test.cc)

add_subdirectory(torch)
Loading