Skip to content

Commit

Permalink
Fix remaining ctests
Browse files Browse the repository at this point in the history
  • Loading branch information
jlebar committed Jun 3, 2024
1 parent 4fadb47 commit a8dfe75
Showing 1 changed file with 10 additions and 3 deletions.
13 changes: 10 additions & 3 deletions unittest/Conversion/TritonGPUToLLVM/EmitIndicesTest.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -1753,7 +1753,14 @@ INSTANTIATE_TEST_SUITE_P(TestCases, LoadSharedToDistributedLLTest,
.shmemCTASplitNum = {1, 1},
.shmemCTAOrder = {1, 0},
.shmemHasLeadingOffset = true,
.shmemStrides = {128, 1},
// The legacy layout code assumes that
// the strides in the row/col
// dimensions are "dense" and match the
// order. LLs can handle other
// strides, but obviously it won't
// match the legacy code, so we can't
// test it here.
.shmemStrides = {1, 128},
.dst =
BlockedLLTestParams{
.sizePerThread = {1, 1},
Expand Down Expand Up @@ -1797,7 +1804,7 @@ INSTANTIATE_TEST_SUITE_P(TestCases, LoadSharedToDistributedLLTest,
.shmemCTASplitNum = {1, 1},
.shmemCTAOrder = {1, 0},
.shmemHasLeadingOffset = true,
.shmemStrides = {128, 1},
.shmemStrides = {1, 128},
.dst =
BlockedLLTestParams{
.sizePerThread = {1, 1},
Expand All @@ -1819,7 +1826,7 @@ INSTANTIATE_TEST_SUITE_P(TestCases, LoadSharedToDistributedLLTest,
.shmemCTASplitNum = {1, 1},
.shmemCTAOrder = {1, 0},
.shmemHasLeadingOffset = true,
.shmemStrides = {128, 1},
.shmemStrides = {1, 128},
.dst =
BlockedLLTestParams{
.sizePerThread = {1, 1},
Expand Down

0 comments on commit a8dfe75

Please sign in to comment.