Support for dynamic slice in generic pattern matching for FP8 GEMMs.

Intel-tensorflow · Aug 25, 2023 · 3e96679 · 3e96679
1 parent 96f9557
commit 3e96679
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tensorflow/compiler/xla/service/gpu/tests/gemm_rewrite_test.cc b/tensorflow/compiler/xla/service/gpu/tests/gemm_rewrite_test.cc
@@ -5001,6 +5001,7 @@ TEST_P(ParameterizedFp8GemmRewriteTest, ScaledABUnscaledDDynamicSliceF8) {
   TF_ASSERT_OK_AND_ASSIGN(bool changed, this->RunHloPass(&pass, module.get()));
   EXPECT_TRUE(changed);
 
+  CheckFp8IfSupported(hlo_text);
   RunAndFilecheckHloRewrite(hlo_text,
                             GemmRewriter(se::CudaComputeCapability{
                                 se::CudaComputeCapability::HOPPER, 0}),