Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
CMakeLists.txt 3.2 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)
Illia Silin's avatar
Split the instances by architecture. (#1223)  
Illia Silin committed
7 
add_example_dependencies(example_contraction_bilinear example_contraction_bilinear_xdl_fp32)
Bartlomiej Wroblewski's avatar
Add support for mixed precision in contraction scale and bilinear (#973)  
Bartlomiej Wroblewski committed
8
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)
Illia Silin's avatar
Split the instances by architecture. (#1223)  
Illia Silin committed
10 
add_example_dependencies(example_contraction_scale example_contraction_scale_xdl_fp32)
Bartlomiej Wroblewski's avatar
Add support for mixed precision in contraction scale and bilinear (#973)  
Bartlomiej Wroblewski committed
11
12 

add_example_executable(example_contraction_bilinear_xdl_fp32_compute_bf16 contraction_bilinear_xdl_fp32_compute_bf16.cpp)
Illia Silin's avatar
Split the instances by architecture. (#1223)  
Illia Silin committed
13 
add_example_dependencies(example_contraction_bilinear example_contraction_bilinear_xdl_fp32_compute_bf16)
Bartlomiej Wroblewski's avatar
Add support for mixed precision in contraction scale and bilinear (#973)  
Bartlomiej Wroblewski committed
14
15 

add_example_executable(example_contraction_scale_xdl_fp32_compute_bf16 contraction_scale_xdl_fp32_compute_bf16.cpp)
Illia Silin's avatar
Split the instances by architecture. (#1223)  
Illia Silin committed
16 
add_example_dependencies(example_contraction_scale example_contraction_scale_xdl_fp32_compute_bf16)
Bartlomiej Wroblewski's avatar
Add support for mixed precision in contraction scale and bilinear (#973)  
Bartlomiej Wroblewski committed
17
18 

add_example_executable(example_contraction_bilinear_xdl_fp32_compute_fp16 contraction_bilinear_xdl_fp32_compute_fp16.cpp)
Illia Silin's avatar
Split the instances by architecture. (#1223)  
Illia Silin committed
19 
add_example_dependencies(example_contraction_bilinear example_contraction_bilinear_xdl_fp32_compute_fp16)
Bartlomiej Wroblewski's avatar
Add support for mixed precision in contraction scale and bilinear (#973)  
Bartlomiej Wroblewski committed
20
21 

add_example_executable(example_contraction_scale_xdl_fp32_compute_fp16 contraction_scale_xdl_fp32_compute_fp16.cpp)
Illia Silin's avatar
Split the instances by architecture. (#1223)  
Illia Silin committed
22 
add_example_dependencies(example_contraction_scale example_contraction_scale_xdl_fp32_compute_fp16)
Bartlomiej Wroblewski's avatar
Add support for mixed precision in contraction scale and bilinear (#973)  
Bartlomiej Wroblewski committed
23
24 

# 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)
Illia Silin's avatar
Split the instances by architecture. (#1223)  
Illia Silin committed
26 
add_example_dependencies(example_contraction_bilinear example_contraction_bilinear_xdl_fp64)
Bartlomiej Wroblewski's avatar
Add support for mixed precision in contraction scale and bilinear (#973)  
Bartlomiej Wroblewski committed
27
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)
Illia Silin's avatar
Split the instances by architecture. (#1223)  
Illia Silin committed
29 
add_example_dependencies(example_contraction_scale example_contraction_scale_xdl_fp64)
Bartlomiej Wroblewski's avatar
Add support for mixed precision in contraction scale and bilinear (#973)  
Bartlomiej Wroblewski committed
30
31 

add_example_executable(example_contraction_bilinear_xdl_fp64_compute_fp32 contraction_bilinear_xdl_fp64_compute_fp32.cpp)
Illia Silin's avatar
Split the instances by architecture. (#1223)  
Illia Silin committed
32 
add_example_dependencies(example_contraction_bilinear example_contraction_bilinear_xdl_fp64_compute_fp32)
Bartlomiej Wroblewski's avatar
Add support for mixed precision in contraction scale and bilinear (#973)  
Bartlomiej Wroblewski committed
33
34 

add_example_executable(example_contraction_scale_xdl_fp64_compute_fp32 contraction_scale_xdl_fp64_compute_fp32.cpp)
Illia Silin's avatar
Split the instances by architecture. (#1223)  
Illia Silin committed
35 
add_example_dependencies(example_contraction_scale example_contraction_scale_xdl_fp64_compute_fp32)
Bartlomiej Wroblewski's avatar
Add support for mixed precision in contraction scale and bilinear (#973)  
Bartlomiej Wroblewski committed
36
37
38 

# FP16
add_example_executable(example_contraction_bilinear_xdl_fp16_compute_fp32 contraction_bilinear_xdl_fp16_compute_fp32.cpp)
Illia Silin's avatar
Split the instances by architecture. (#1223)  
Illia Silin committed
39 
add_example_dependencies(example_contraction_bilinear example_contraction_bilinear_xdl_fp16_compute_fp32)
Bartlomiej Wroblewski's avatar
Add support for mixed precision in contraction scale and bilinear (#973)  
Bartlomiej Wroblewski committed
40
41 

add_example_executable(example_contraction_scale_xdl_fp16_compute_fp32 contraction_scale_xdl_fp16_compute_fp32.cpp)
Illia Silin's avatar
Split the instances by architecture. (#1223)  
Illia Silin committed
42 
add_example_dependencies(example_contraction_scale example_contraction_scale_xdl_fp16_compute_fp32)
Bartlomiej Wroblewski's avatar
Add support for mixed precision in contraction scale and bilinear (#973)  
Bartlomiej Wroblewski committed
43
44
45 

# BF16
add_example_executable(example_contraction_bilinear_xdl_bf16_compute_fp32 contraction_bilinear_xdl_bf16_compute_fp32.cpp)
Illia Silin's avatar
Split the instances by architecture. (#1223)  
Illia Silin committed
46 
add_example_dependencies(example_contraction_bilinear example_contraction_bilinear_xdl_bf16_compute_fp32)
Bartlomiej Wroblewski's avatar
Add support for mixed precision in contraction scale and bilinear (#973)  
Bartlomiej Wroblewski committed
47
48 

add_example_executable(example_contraction_scale_xdl_bf16_compute_fp32 contraction_scale_xdl_bf16_compute_fp32.cpp)
Illia Silin's avatar
Split the instances by architecture. (#1223)  
Illia Silin committed
49 
add_example_dependencies(example_contraction_scale example_contraction_scale_xdl_bf16_compute_fp32)
Bartlomiej Wroblewski's avatar
Add support for mixed precision in contraction scale and bilinear (#973)  
Bartlomiej Wroblewski committed
50
Illia Silin's avatar
Split the instances by architecture. (#1223)  
Illia Silin committed
51
52 
add_example_dependencies(example_contraction example_contraction_scale)
add_example_dependencies(example_contraction example_contraction_bilinear)