forked from PaddlePaddle/Paddle
-
Notifications
You must be signed in to change notification settings - Fork 1
Expand file tree
/
Copy pathCMakeLists.txt
More file actions
93 lines (87 loc) · 4.41 KB
/
CMakeLists.txt
File metadata and controls
93 lines (87 loc) · 4.41 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
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)
# 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)
# 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()
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(ATen_pin_memory_kernel_test SRCS ATen_pin_memory_kernel_test.cu)
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)
endif()
add_subdirectory(torch)