mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-19 20:40:07 +00:00
7d896f7d0d
* 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>
[ROCm/composable_kernel commit: 4ef704d8a6]
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)
|