Skip to content

Commit 3e8f256

Browse files
committed
Disable subtests failing on mi250
1 parent f1bbc43 commit 3e8f256

1 file changed

Lines changed: 8 additions & 0 deletions

File tree

tensorflow/tools/ci_build/linux/rocm/run_xla.sh

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -65,6 +65,7 @@ EXCLUDED_TESTS=(
6565
CompareTest.SplitKBatch
6666
CompareTest.SupportsSplitKWithIndivisibleKUsingPaddingEqual1
6767
TritonGemmTest.S8ToF32DotWithManyWarpsDoesNotCrash
68+
TritonGemmTest.SplitAndTransposeLhsExecutesCorrectly #failing on mi250
6869

6970
# @xla//xla/backends/gpu/codegen/triton:fusion_emitter_device_test_amdgpu_any
7071
TritonEmitterTest.FusionWithOutputContainingMoreThanInt32MaxElementsExecutesCorrectly
@@ -107,6 +108,10 @@ EXCLUDED_TESTS=(
107108
TritonFusionNumericsVerifierTestSuite/TritonFusionNumericsVerifierTest.VerifyMultipleNestedFusionNumerics/0
108109
TritonFusionNumericsVerifierTestSuite/TritonFusionNumericsVerifierTest.VerifyMultipleNestedFusionNumerics/1
109110

111+
# @xla//xla/service/gpu/transforms:dot_dimension_sorter_test_amdgpu_any
112+
WithoutDotDimensionSorterTest.UnsortedDimsCreateTransposes # failing on mi250
113+
WithoutDotDimensionSorterTest.DimOrderCanBeChanged # failing on mi250
114+
110115
# @xla//xla/service/gpu/tests:gpu_cub_sort_test_amdgpu_any
111116
CubSortKeysTest.CompareToReferenceNumpyOrderGt
112117
CubSortKeysTest.CompareToReferenceTotalOrderLt
@@ -221,6 +226,9 @@ EXCLUDED_TESTS=(
221226
# @xla//xla/tools:xla_deviceless_compile_lib_test
222227
XlaDevicelessCompileLibTest/XlaDevicelessCompileLibTest.CompilesForGpuWithoutDevice/LegacyAotFlow
223228
XlaDevicelessCompileLibTest/XlaDevicelessCompileLibTest.CompilesForGpuWithoutDevice/NewAotFlow
229+
230+
# @xla//xla/service/gpu/autotuning:gemm_fusion_autotuner_test_amdgpu_any
231+
GemmFusionAutotunerTest.Int8FusedGemm # failing on mi250
224232
)
225233

226234
bazel --bazelrc=tensorflow/tools/tf_sig_build_dockerfiles/devel.usertools/rocm.bazelrc test \

0 commit comments

Comments
 (0)