Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
CMakeLists.txt 3.07 KB
Newer Older
Bartlomiej Wroblewski's avatar
Add support for mixed precision in contraction scale and bilinear (#973)  
Bartlomiej Wroblewski committed
1
2
3
4
5 
add_custom_target(example_contraction)
add_custom_target(example_contraction_scale)
add_custom_target(example_contraction_bilinear)

# FP32
Illia Silin's avatar
Refactoring cmake files to build data types separately. (#932)  
Illia Silin committed
6 
add_example_executable(example_contraction_bilinear_xdl_fp32 contraction_bilinear_xdl_fp32.cpp)
Bartlomiej Wroblewski's avatar
Add support for mixed precision in contraction scale and bilinear (#973)  
Bartlomiej Wroblewski committed
7
8 
add_dependencies(example_contraction_bilinear example_contraction_bilinear_xdl_fp32)
Illia Silin's avatar
Refactoring cmake files to build data types separately. (#932)  
Illia Silin committed
9 
add_example_executable(example_contraction_scale_xdl_fp32 contraction_scale_xdl_fp32.cpp)
Bartlomiej Wroblewski's avatar
Add support for mixed precision in contraction scale and bilinear (#973)  
Bartlomiej Wroblewski committed
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24 
add_dependencies(example_contraction_scale example_contraction_scale_xdl_fp32)

add_example_executable(example_contraction_bilinear_xdl_fp32_compute_bf16 contraction_bilinear_xdl_fp32_compute_bf16.cpp)
add_dependencies(example_contraction_bilinear example_contraction_bilinear_xdl_fp32_compute_bf16)

add_example_executable(example_contraction_scale_xdl_fp32_compute_bf16 contraction_scale_xdl_fp32_compute_bf16.cpp)
add_dependencies(example_contraction_scale example_contraction_scale_xdl_fp32_compute_bf16)

add_example_executable(example_contraction_bilinear_xdl_fp32_compute_fp16 contraction_bilinear_xdl_fp32_compute_fp16.cpp)
add_dependencies(example_contraction_bilinear example_contraction_bilinear_xdl_fp32_compute_fp16)

add_example_executable(example_contraction_scale_xdl_fp32_compute_fp16 contraction_scale_xdl_fp32_compute_fp16.cpp)
add_dependencies(example_contraction_scale example_contraction_scale_xdl_fp32_compute_fp16)

# FP64
Illia Silin's avatar
Refactoring cmake files to build data types separately. (#932)  
Illia Silin committed
25 
add_example_executable(example_contraction_bilinear_xdl_fp64 contraction_bilinear_xdl_fp64.cpp)
Bartlomiej Wroblewski's avatar
Add support for mixed precision in contraction scale and bilinear (#973)  
Bartlomiej Wroblewski committed
26
27 
add_dependencies(example_contraction_bilinear example_contraction_bilinear_xdl_fp64)
Illia Silin's avatar
Refactoring cmake files to build data types separately. (#932)  
Illia Silin committed
28 
add_example_executable(example_contraction_scale_xdl_fp64 contraction_scale_xdl_fp64.cpp)
Bartlomiej Wroblewski's avatar
Add support for mixed precision in contraction scale and bilinear (#973)  
Bartlomiej Wroblewski committed
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52 
add_dependencies(example_contraction_scale example_contraction_scale_xdl_fp64)

add_example_executable(example_contraction_bilinear_xdl_fp64_compute_fp32 contraction_bilinear_xdl_fp64_compute_fp32.cpp)
add_dependencies(example_contraction_bilinear example_contraction_bilinear_xdl_fp64_compute_fp32)

add_example_executable(example_contraction_scale_xdl_fp64_compute_fp32 contraction_scale_xdl_fp64_compute_fp32.cpp)
add_dependencies(example_contraction_scale example_contraction_scale_xdl_fp64_compute_fp32)

# FP16
add_example_executable(example_contraction_bilinear_xdl_fp16_compute_fp32 contraction_bilinear_xdl_fp16_compute_fp32.cpp)
add_dependencies(example_contraction_bilinear example_contraction_bilinear_xdl_fp16_compute_fp32)

add_example_executable(example_contraction_scale_xdl_fp16_compute_fp32 contraction_scale_xdl_fp16_compute_fp32.cpp)
add_dependencies(example_contraction_scale example_contraction_scale_xdl_fp16_compute_fp32)

# BF16
add_example_executable(example_contraction_bilinear_xdl_bf16_compute_fp32 contraction_bilinear_xdl_bf16_compute_fp32.cpp)
add_dependencies(example_contraction_bilinear example_contraction_bilinear_xdl_bf16_compute_fp32)

add_example_executable(example_contraction_scale_xdl_bf16_compute_fp32 contraction_scale_xdl_bf16_compute_fp32.cpp)
add_dependencies(example_contraction_scale example_contraction_scale_xdl_bf16_compute_fp32)

add_dependencies(example_contraction example_contraction_scale)
add_dependencies(example_contraction example_contraction_bilinear)