Work around develop validation failure (#513)

* workaround bf16 atten fwd issue on gfx908

* typo

[ROCm/composable_kernel commit: 892a8d769d]
This commit is contained in:
Anthony Chang
2022-11-18 00:38:13 +08:00
committed by GitHub
parent 0015ce6288
commit 683194f86a
3 changed files with 16 additions and 1 deletions

View File

@@ -29,7 +29,7 @@ TYPED_TEST_SUITE(TestBatchedGemmMaskingScaleSoftmaxGemmPermuteBF16, KernelTypes)
TYPED_TEST(TestBatchedGemmMaskingScaleSoftmaxGemmPermuteBF16, Test_BF16) { this->Run(); }
TYPED_TEST(TestBatchedGemmMaskingScaleSoftmaxGemmPermuteBF16, Test_FPBF_PadM)
TYPED_TEST(TestBatchedGemmMaskingScaleSoftmaxGemmPermuteBF16, Test_BF16_PadM)
{
this->lengths_ = std::vector<std::vector<int>>{
{136, 128, 32, 128, 2, 3},