ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-16 02:54:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

GemmGemm TNNT instances (#399)

Browse Source 
* add gemm_gemm TNNT instance

* sanitize Gemm1KPack

* disable instances that failed validation on mi100

[ROCm/composable_kernel commit: fe52c94c98]
This commit is contained in:
Anthony Chang
2022-09-07 02:38:01 +08:00
committed by GitHub
parent 5643625481
commit 2d119fda7b
6 changed files with 107 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
test/batched_gemm_gemm/test_batched_gemm_gemm_fp16.cpp
View File

@@ -11,7 +11,8 @@ class TestBatchedGemmGemmFP16 : public TestBatchedGemmGemm<Tuple>
// clang-format off
using KernelTypes = ::testing::Types<
std::tuple<F16, F16, F16, F16, Row, Col, Row, Row>
std::tuple<F16, F16, F16, F16, Row, Col, Row, Row>,
std::tuple<F16, F16, F16, F16, Row, Col, Col, Row>
>;
// clang-format on