Commit 552c6b23 authored by Po-Yen, Chen's avatar Po-Yen, Chen
Browse files

Add instance for layout=TT

parent 82a7b60d
...@@ -295,7 +295,7 @@ struct DeviceOperationInstanceFactory< ...@@ -295,7 +295,7 @@ struct DeviceOperationInstanceFactory<
if constexpr(is_same_v<ALayout, Row> && is_same_v<BLayout, Row> && if constexpr(is_same_v<ALayout, Row> && is_same_v<BLayout, Row> &&
is_same_v<CLayout, Row>) is_same_v<CLayout, Row>)
{ {
// add_device_gemm_xdl_f16_f16_f16_mk_kn_mn_instances(op_ptrs); add_device_gemm_xdl_f16_f16_f16_mk_kn_mn_instances(op_ptrs);
// add_device_gemm_dl_f16_f16_f16_mk_kn_mn_instances(op_ptrs); // add_device_gemm_dl_f16_f16_f16_mk_kn_mn_instances(op_ptrs);
// add_device_gemm_xdl_c_shuffle_f16_f16_f16_mk_kn_mn_instances(op_ptrs); // add_device_gemm_xdl_c_shuffle_f16_f16_f16_mk_kn_mn_instances(op_ptrs);
} }
......
...@@ -7,7 +7,7 @@ add_instance_library(device_gemm_instance ...@@ -7,7 +7,7 @@ add_instance_library(device_gemm_instance
# device_gemm_xdl_f32_f32_f32_mk_nk_mn_instance.cpp # device_gemm_xdl_f32_f32_f32_mk_nk_mn_instance.cpp
# device_gemm_xdl_f32_f32_f32_km_kn_mn_instance.cpp # device_gemm_xdl_f32_f32_f32_km_kn_mn_instance.cpp
# device_gemm_xdl_f32_f32_f32_km_nk_mn_instance.cpp # device_gemm_xdl_f32_f32_f32_km_nk_mn_instance.cpp
# device_gemm_xdl_f16_f16_f16_mk_kn_mn_instance.cpp device_gemm_xdl_f16_f16_f16_mk_kn_mn_instance.cpp
device_gemm_xdl_f16_f16_f16_mk_nk_mn_instance.cpp device_gemm_xdl_f16_f16_f16_mk_nk_mn_instance.cpp
# device_gemm_xdl_f16_f16_f16_km_kn_mn_instance.cpp # device_gemm_xdl_f16_f16_f16_km_kn_mn_instance.cpp
# device_gemm_xdl_f16_f16_f16_km_nk_mn_instance.cpp # device_gemm_xdl_f16_f16_f16_km_nk_mn_instance.cpp
...@@ -44,6 +44,11 @@ add_instance_library(device_gemm_instance ...@@ -44,6 +44,11 @@ add_instance_library(device_gemm_instance
set(ENABLE_IGLP_OPT ON) set(ENABLE_IGLP_OPT ON)
if(ENABLE_IGLP_OPT) if(ENABLE_IGLP_OPT)
# layout=TT
set_source_files_properties(device_gemm_xdl_f16_f16_f16_mk_kn_mn_instance.cpp PROPERTIES
COMPILE_OPTIONS ";--save-temps;-Wno-gnu-line-marker;"
COMPILE_DEFINITIONS ";USE_WAVES_PER_EU;")
# layout=TN
set_source_files_properties(device_gemm_xdl_f16_f16_f16_mk_nk_mn_instance.cpp PROPERTIES set_source_files_properties(device_gemm_xdl_f16_f16_f16_mk_nk_mn_instance.cpp PROPERTIES
COMPILE_OPTIONS ";--save-temps;-Wno-gnu-line-marker;-mllvm;-amdgpu-enable-max-ilp-scheduling-strategy;" COMPILE_OPTIONS ";--save-temps;-Wno-gnu-line-marker;-mllvm;-amdgpu-enable-max-ilp-scheduling-strategy;"
COMPILE_DEFINITIONS ";USE_IGLP_OPT;USE_WAVES_PER_EU;") COMPILE_DEFINITIONS ";USE_IGLP_OPT;USE_WAVES_PER_EU;")
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment