ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-06-08 15:30:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add missing ComputeDatatype in contraction_multi_ABD_xdl_fp16 (#1024)

Browse Source
This commit is contained in:
Bartlomiej Wroblewski
2023-11-03 16:22:11 +01:00
committed by GitHub
parent 4ef704d8a6
commit 16eb824c90
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
example/61_contraction_multi_ABD/contraction_multi_ABD_xdl_fp16.cpp
View File

@@ -34,6 +34,7 @@ using AccDataType = F32;
using CShuffleDataType = F32;
using DDataType = F16;
using EDataType = F16;
using ComputeDataType = F16;
static constexpr ck::index_t NumDimM = 2;
static constexpr ck::index_t NumDimN = 2;
@@ -291,6 +292,7 @@ int main(int argc, char* argv[])
BDataType,
CShuffleDataType,
AccDataType,
ComputeDataType,
PassThrough,
BElementOp>;