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

Fix continous dim selection in contraction (#1336)

Browse Source 
* Fix continous dim selection in contraction

* Fixes
This commit is contained in:
Bartłomiej Kocot
2024-06-18 10:26:49 +02:00
committed by GitHub
parent 17ed368f58
commit 933951ed48
4 changed files with 80 additions and 59 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
test/contraction/test_contraction_xdl.cpp
View File

@@ -212,4 +212,10 @@ TYPED_TEST(TestContractionScaleMixedPrecision, scale)
this->template Run<6>({{1, 1, 1, 3, 2, 3}, {1, 1, 1, 3, 2, 3}, {1, 1, 1, 2, 2, 4}});
this->template Run<2>({{16, 8}, {16, 8}, {16, 8}});
this->template Run<2>({{8, 16}, {16, 8}, {8, 16}});
// special cases
this->template Run<2>({{1, 1}, {16, 8}, {8, 16}});
this->template Run<2>({{8, 16}, {16, 8}, {1, 1}});
this->template Run<2>({{8, 16}, {1, 1}, {8, 16}});
this->template Run<2>({{1, 1}, {1, 1}, {1, 1}});
}