mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-11 17:00:18 +00:00
4ef704d8a6
* Add support for mixed precision in contraction scale and bilinear (#936) * Extract common functionality to separate files * Reference contraction: Remove incorrect consts from type_converts * Reference contraction: Add missing type_convert for dst value * Reference contraction: Fix incorrect order of B matrix dimensions * Add support for mixed precision in contraction scale and bilinear * Move using statements from instances to a common file * Move using statements from examples to a common file * Fix the order of B matrix dimensions across examples and profiler * Fix the computation of error threshold * Make ComputeDataType an optional argument * Include possible DataType -> ComputeDataType casting error in the threshold * Remove commented code * Make the ComputeDataType an optional argument in instance --------- Co-authored-by: Illia Silin <98187287+illsilin@users.noreply.github.com>
53 lines
3.1 KiB
CMake
53 lines
3.1 KiB
CMake
add_custom_target(example_contraction)
|
|
add_custom_target(example_contraction_scale)
|
|
add_custom_target(example_contraction_bilinear)
|
|
|
|
# FP32
|
|
add_example_executable(example_contraction_bilinear_xdl_fp32 contraction_bilinear_xdl_fp32.cpp)
|
|
add_dependencies(example_contraction_bilinear example_contraction_bilinear_xdl_fp32)
|
|
|
|
add_example_executable(example_contraction_scale_xdl_fp32 contraction_scale_xdl_fp32.cpp)
|
|
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
|
|
add_example_executable(example_contraction_bilinear_xdl_fp64 contraction_bilinear_xdl_fp64.cpp)
|
|
add_dependencies(example_contraction_bilinear example_contraction_bilinear_xdl_fp64)
|
|
|
|
add_example_executable(example_contraction_scale_xdl_fp64 contraction_scale_xdl_fp64.cpp)
|
|
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)
|