From 614c0c05cfa947f2d9e4053df24a9d7a52710d87 Mon Sep 17 00:00:00 2001 From: Harsha H S Date: Mon, 16 Sep 2024 04:02:08 -0700 Subject: [PATCH] PR #17156: [ROCm] Skip ConditionalIfWithMemset on ROCm, introduced in `201df9ca00` Imported from GitHub PR https://github.com/openxla/xla/pull/17156 Copybara import of the project: -- 2849f5297b9685fdff1a41a342751d25b4113839 by Harsha HS : [ROCm] Skip ConditionalIfWithMemset on ROCm, introduced in `201df9ca00` Merging this change closes #17156 COPYBARA_INTEGRATE_REVIEW=https://github.com/openxla/xla/pull/17156 from ROCm:ci_skip_ConditionalIfWithMemset_onROCm_20240913 2849f5297b9685fdff1a41a342751d25b4113839 PiperOrigin-RevId: 675087448 --- xla/stream_executor/gpu/gpu_command_buffer_test.cc | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/xla/stream_executor/gpu/gpu_command_buffer_test.cc b/xla/stream_executor/gpu/gpu_command_buffer_test.cc index 4ec9900821957..aaf2402b6e829 100644 --- a/xla/stream_executor/gpu/gpu_command_buffer_test.cc +++ b/xla/stream_executor/gpu/gpu_command_buffer_test.cc @@ -783,6 +783,10 @@ TEST(GpuCommandBufferTest, ConditionalIfWithMemset) { Platform* platform = GpuPlatform(); StreamExecutor* executor = platform->ExecutorForDevice(0).value(); + if (platform->id() == rocm::kROCmPlatformId) { + GTEST_SKIP() << "Not supported on ROCM"; + } + if (platform->id() == cuda::kCudaPlatformId && executor->GetDeviceDescription().driver_version() < SemanticVersion{12, 4, 0}) {