From 3b481df87278acbb5a6d2d0b82b4f95b590d14db Mon Sep 17 00:00:00 2001
From: jayfurmanek <Jason.Furmanek@amd.com>
Date: Tue, 10 Sep 2024 08:19:26 -0500
Subject: [PATCH] Disable kAllNHWC (#2655)

---
 third_party/xla/xla/service/gpu/gpu_layout_assignment.cc | 4 ----
 1 file changed, 4 deletions(-)

diff --git a/third_party/xla/xla/service/gpu/gpu_layout_assignment.cc b/third_party/xla/xla/service/gpu/gpu_layout_assignment.cc
index b260353292de47..35707685af2741 100644
--- a/third_party/xla/xla/service/gpu/gpu_layout_assignment.cc
+++ b/third_party/xla/xla/service/gpu/gpu_layout_assignment.cc
@@ -129,10 +129,6 @@ HeuristicLayoutAssignment(const HloInstruction* instr,
     return kAllNHWC;
   }
 
-  const auto* rocm_compute_capability =
-      std::get_if<se::RocmComputeCapability>(&gpu_version);
-  if (rocm_compute_capability && input_ty == F16) return kAllNHWC;
-
   // If we're not Volta or not fp16/bfloat16, or not conv2D, the decision is
   // easy: Use NCHW.
   const bool isFloat16 = (input_ty == F16) || (input_ty == BF16);