Skip to content

GitLab

"megatron/legacy/model/bert_model.py" did not exist on "ebbe40cd3647168c808c8f877355e8d5ceb5ef20"
Switch branch/tag
Find file Normal viewHistoryPermalink
CMakeLists.txt 744 Bytes
Newer Older
Adam Osewski's avatar
Unified conv3D API + support for all data types. (#133)  
Adam Osewski committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17 
add_custom_target(test_convnd_fwd)

add_test_executable(test_conv1d_fwd conv1d_fwd.cpp)
target_link_libraries(test_conv1d_fwd PRIVATE host_tensor)
target_link_libraries(test_conv1d_fwd PRIVATE device_conv1d_fwd_instance)
add_dependencies(test_convnd_fwd test_conv1d_fwd)

add_test_executable(test_conv2d_fwd conv2d_fwd.cpp)
target_link_libraries(test_conv2d_fwd PRIVATE host_tensor)
target_link_libraries(test_conv2d_fwd PRIVATE device_conv2d_fwd_instance)
add_dependencies(test_convnd_fwd test_conv2d_fwd)

add_test_executable(test_conv3d_fwd conv3d_fwd.cpp)
target_link_libraries(test_conv3d_fwd PRIVATE host_tensor)
target_link_libraries(test_conv3d_fwd PRIVATE device_conv3d_fwd_instance)
add_dependencies(test_convnd_fwd test_conv3d_fwd)