ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-04-20 14:59:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

[Hotfix] SplitK Gemm fp32 (#401)

Browse Source 
* add scripts

* fixed splitK_gemm_fp32

* clean

* clean

* use gemm_xdl_splitK_c_shuffle into profiler

* remove device_gemm_xdl_splitk.hpp
This commit is contained in:
zjing14
2022-09-02 11:16:09 -05:00
committed by GitHub
parent 204ef976ca
commit 7589116121
6 changed files with 58 additions and 710 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
test/gemm_split_k/gemm_split_k.cpp
View File

@@ -8,7 +8,6 @@
#include "ck/ck.hpp"
#include "ck/tensor_operation/gpu/device/tensor_layout.hpp"
#include "ck/tensor_operation/gpu/device/gemm_specialization.hpp"
#include "ck/tensor_operation/gpu/device/device_gemm_xdl_splitk.hpp"
#include "ck/tensor_operation/gpu/element/element_wise_operation.hpp"
#include "ck/library/tensor_operation_instance/gpu/gemm_splitk.hpp"